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)
tensorflow预训练参数解析
最新推荐文章于 2019-11-10 09:51:23 发布