Tensorflow_LSTM图片测试batch_size的设置问题
1.书写本篇博客的目的首先是为了让自己不要忘了,顺便帮助遇到一样困扰的小伙伴而已,文笔不佳,请见谅。
2.进入正题:在使用LSTM网络做图片分类测试的时候
注意一个参数:batch_size
在定义LSTM_cell的时候:lstm_cell = tf.nn.rnn_cell.BasicLSTMCell(n_hidden_units, forget_bias=1.0, state_is_t...
原创
2019-05-10 16:41:08 ·
2639 阅读 ·
0 评论