Checkpoint a bit more often.

diff --git a/voxel_flow_train.py b/voxel_flow_train.py
index 736255b184f41909de9f3c071acf6027ec1e8823..0ba31722b156a8ccb08d6e3cd0205d15992fd06a 100755 (executable)
--- a/voxel_flow_train.py
+++ b/voxel_flow_train.py
@@ -160,7 +160,7 @@ def train(dataset_frame1, dataset_frame2, dataset_frame3):
           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)