查看TensorFlow checkpoint文件中的变量名和对应值方法
实例如下所示:
from tensorflow.python import pywrap_tensorflow checkpoint_path = os.path.join(model_dir, "model.ckpt") reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path) var_to_shape_map = reader.get_variable_to_shape_map() for key in var_to_shape_map: print("tensor_name: ", key) print(reader.get_tensor(key)) # Remove this is you want to print only variable names
相关推荐
暗夜之城 2020-05-07
silencehgt 2020-05-07
woshigzp 2020-01-21
daixiang 2020-01-21
WSNjiang 2020-01-21
jianghuchuanke 2019-11-17
louishao 2019-10-20
xiaoyutongxue 2019-08-16
eternityzzy 2013-05-21
aliushuo 2011-03-06
Jinyli 2019-06-29
农村外出务工男 2019-06-21
ououlal 2012-11-13
xorxos 2018-11-08
亚恋之欣 2019-03-25
budding0 2016-05-06
aotou 2012-11-07
PostgreSql 2019-02-24