方法 | 返回值类型 | 参数 | 说明 |
.train.Saver() | 实例s | var_list=None | 指定被保存和恢复的变量 |
dict: {name: 变量} | |||
list: [变量] | |||
None: 所有saveable对象列表 | |||
reshape=False | 是否允许更改变量的形状 | ||
sharded=False | |||
max_to_keep=5 | |||
keep_checkpoint_ every_n_hours=10000.0 | |||
name=None | 操作名 | ||
restore_sequentially=F | |||
saver_def=None | |||
builder=None | |||
defer_build=False | |||
allow_empty=False | |||
write_version= tf.train.SaverDef.V2 | |||
pad_step_number=False | |||
save_relative_paths=F | |||
filename=None | |||
s.restore() | 恢复以前保存的变量(需定义好, 和保存的变量形状dtype一致, 不需要初始化) | ||
sess | 需要变量回复到的session | ||
save_path | 路径 | ||
s.save() | str(path) | 保存变量 | |
sess | session(保存其中所有变量) | ||
save_path | 保存到的路径及文件名 | ||
global_step=None | |||
latest_filename=None | |||
meta_graph_suffix='meta' | |||
write_meta_graph=True | |||
write_state=True | |||