欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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)

比较有什么不同,然后改成一致,就可以加载

相关标签: tensorflow