tf.train.Supervisor
tf.train.Supervisor(logdir=log_path, init_op=init)可以简化编程,避免显示地实现restore操作.其会判断模型是否存在.如果存在,会自动读取模型.不用显式地调用restore.managed_session()
sv=tf.train.Supervisor(logdir=log_path, init_op=init)
with sv.managed_session() as sess: #会自动去logdir中去找checkpoint,如果没有的话,自动执行初始化
TensorFlow函数使用方法记录
最新推荐文章于 2023-12-03 17:03:58 发布