加载张量流CUDNNRNNRelu检查点以进行cpu推理

时间:2018-03-13 10:32:43

标签: tensorflow

我目前正在使用张量流上的CudnnRNNRelu训练双向RNN模型。我想在CPU机器上使用这些检查点进行推理。文档表明,使用平台无关的rnn单元(tf.contrib.rnn.BasicRNNCell等)无法直接恢复使用Cudnn特定细胞(CudnnRNNRelu,CudnnRNNTanh,CudnnLSTM等)训练的模型。虽然有单独的细胞来恢复LSTM和GRU细胞(CudnnComaptibleLSTMCell和CudnnCompatibleGRUCell),但没有迹象表明如何将CudnnRNNRelu训练的检查点与平台无关的rnn细胞一起使用。

恢复在GPU计算机上正常运行。有没有办法恢复CudnnRNNRelu检查站?

0 个答案:

没有答案