TensorFlow模型恢复小技巧

TensorFlow下需要查看张量参数。

import os
from tensorflow.python import pywrap_tensorflow
reader = pywrap_tensorflow.NewCheckpointReader('/usr/local/lib/python3.7/dist-packages/chicksexer/models/model.ckpt') #替换为自己的路径
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
    print("tensor_name: ", key)

参考链接：
https://blog.csdn.net/weixin_44393192/article/details/108513005
https://blog.csdn.net/qq_36810544/article/details/81456182

本文链接：