tensorflow查看所有变量和checkpoint保存的变量
程序员文章站
2022-03-31 19:51:58
...
当tensorflow加载训练好的模型时,如果现在创建的图中变量与保存的不一致时,就会报错,报出类似这样的错误
NotFoundError (see above for traceback): Key Variable not found in checkpoint [[Node: save/RestoreV2
这时我们可以打印出所有变量名,比较有什么不同
对于checkpoint保存的:
from tensorflow.contrib.framework.python.framework import checkpoint_utils
var_list = checkpoint_utils.list_variables(config.pre_model_dir)
for v in var_list:
print(v)
对于当前全局变量:
for variable_name in tf.global_variables():
print(variable_name)
比较有什么不同,然后改成一致,就可以加载
上一篇: 因癌症而死的原因竟是每天多喝了几杯它
下一篇: 喝茶能防雾霾? “雾霾清肺茶”其实不靠谱