启动训练
关键步骤操作参考如下。
定义Saver用于训练过程中模型的保存和加载,启动Session计算并在训练过程中保存(
tf.compat.v1.train.Saver.save
)或加载(
tf.compat.v1.train.Saver.restore
)模型。
启动训练任务。可参见
单机单卡和单机多卡训练
。
父主题:
使用tf.Session训练