Tensorflow未保存所需的模型
我要求tensorflow在每个历元中每100次迭代保存一次模型，下面是我的代码。但经过900次迭代后，只保存了第500、600、700、800和900次迭代的训练模型Tensorflow未保存所需的模型,tensorflow,Tensorflow,我要求tensorflow在每个历元中每100次迭代保存一次模型，下面是我的代码。但经过900次迭代后，只保存了第500、600、700、800和900次迭代的训练模型 with tf.Session(config = tf.ConfigProto(log_device_placement = True)) as sess: sess.run(init_op) for i in range(args.num_epochs): start_time = time
with tf.Session(config = tf.ConfigProto(log_device_placement = True)) as sess:
sess.run(init_op)
for i in range(args.num_epochs):
start_time = time.time()
k = 0
acc_train = 0
# initialize the iterator to train_dataset
sess.run(train_init_op)
while True:
try:
accu, l, _ = sess.run([accuracy, loss, optimizer], feed_dict = {training: True})
k += 1
acc_train += accu
if k % 100 == 0:
print('Epoch: {}, step: {}, training loss: {:.3f}, training accuracy: {:.2f}%'.format(i, k, l, accu * 100))
saver.save(sess, args.saved_model_path, global_step = (i+1) * k)
except tf.errors.OutOfRangeError:
break
我的问题是为什么tensorflow没有为第100、200、300、400次迭代保存模型？谢谢大家! 确实如此，但我猜您创建的实例的默认值为
max\u keepsaver = tf.train.Saver(max_keep=10)
如果不想保存-every-one，您可能还想使用
keep\u checkpoint\u every\n\u hoursmax\u keepsaver = tf.train.Saver(max_keep=10)
如果您不想保存-every-one，您可能还需要使用
keep\u checkpoint\u every\n\u hours