成功解决代码运行出现的问题:
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, 1, 100, 1, 196, 512, 0]
[[{{node CudnnRNN}}]]
[[model_3/bidirectional_1/forward_gru_9/PartitionedCall]] [Op:__inference_train_function_35876]
通过查看经验贴发现时batch_size太大了 ,调小一点就行了。