您的位置:首页 > 其它

tensorflow预训练参数解析

2017-06-16 16:34 183 查看
from tensorflow.python import pywrap_tensorflow

reader=pywrap_tensorflow.NewCheckpointReader(In_ckpt_path)
var_to_shape_map = reader.get_variable_to_shape_map()
featureDict = dict()
for key in var_to_shape_map:
featureDict[key] = reader.get_tensor(key)

print(featureDict)
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: