1. 程式人生 > >Tensorflow報錯tensorflow.python.framework.errors_impl.InvalidArgumentError exception str() failed原因

Tensorflow報錯tensorflow.python.framework.errors_impl.InvalidArgumentError exception str() failed原因

用Tensorflow跑自己的資料集時,用saver將引數儲存在checkpoint中。

後來在在測試的時候,用saver.restore匯入模型到Session中,報如下錯誤:

tensorflow.python.framework.errors_impl.InvalidArgumentError exception str() failed

出現這個錯誤的原因是:我修改了網路的優化方法,在restore時讀取模型導致引數不匹配。

改回之前的優化方法即可

optimizer = tf.train.AdamOptimizer(FLAGS.learning_rate)  # Adam優化演算法.
# optimizer = tf.train.MomentumOptimizer(FLAGS.learning_rate, 0.9) # 0.9為保留的動量方向比例

建議如要修改網路,將原始碼和logs備份