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预训练参数解析
最新推荐文章于 2022-07-11 18:10:08 发布