imwrite(file_name_label, target_np[i,:,:,:])
# Save checkpoint
- if step % 5000 == 0 or (step +1) == FLAGS.max_steps:
+ if step % 500 == 0 or (step +1) == FLAGS.max_steps:
checkpoint_path = os.path.join(FLAGS.train_dir, 'model.ckpt')
saver.save(sess, checkpoint_path, global_step=step)