GPU版TensorFlow预测报错
来源：10-10 样例例分析与总结
朝堂布衣
2021-12-09
老师，请问我用GPU训练，训练过程没问题，最后预测的时候 报这个错，大概是什么原因？
InternalError: Failed to call ThenRnnForward with model config: [rnn_mode, rnn_input_mode, rnn_direction_mode]: 3, 0, 0 , [num_layers, input_size, num_units, dir_count, max_seq_length, batch_size, cell_num_units]: [1, 1280, 1024, 1, 1, 1, 0] [Op:CudnnRNN]
写回答
1回答
-
正十七
2021-12-27
see https://github.com/mozilla/DeepSpeech/issues/3088
https://stackoverflow.com/questions/53972814/cudnnlstm-failed-to-call-thenrnnforward
看起来像是memory不足的问题，可以考虑降低batch_size.
00
相似问题