TensorFlow训练模型及保存数量限制的解决办法

    每次卷积神经网络训练的结果都只保存了最后一部分,查阅了一下相关资料,发现是定义saver时采用的默认值,这里进行如下设置:

saver = tf.train.Saver( max_to_keep = 100 , keep_checkpoint_every_n_hours = 1 )

猜你喜欢

转载自blog.csdn.net/weixin_41863859/article/details/80222306