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