解决TensorFlow模型恢复时的未初始化变量报错

TensorFlow模型恢复报错现象

错误信息

Attempting to use uninitialized value input_producer/input_producer/limit_epochs/epochs

错误原因分析

在尝试恢复模型时遇到上述报错，经排查发现该问题源于使用了tf.train.slice_input_producer方法。该函数内部机制将其视为一个Tensor，若未正确初始化则会导致报错。

解决方案

修改方法：

在获取batch数据之前，需在会话（sess）中先执行初始化操作。

sess.run(tf.global_variables_initializer())
sess.run(tf.local_variables_initializer())

总结

以上这篇解决TensorFlow模型恢复报错的问题就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持本站。