TensorFlow中saver.restore报错

可能造成的原因:预训练的参数少于训练的参数

解决方法:

导入模型保存的参数时只选择预训练过程涉及到的参数,例如:从所有变量中选取与预训练对应的17个variable。

TensorFlow中saver.restore报错_第1张图片

你可能感兴趣的:(Tensor_Flow学习笔记)