X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;ds=sidebyside;f=voxel_flow_train.py;h=0ba31722b156a8ccb08d6e3cd0205d15992fd06a;hb=HEAD;hp=736255b184f41909de9f3c071acf6027ec1e8823;hpb=a36660b4e7c0ddb51c0aa5ad4b2a9aa6cf483349;p=voxel-flow

diff --git a/voxel_flow_train.py b/voxel_flow_train.py
index 736255b..0ba3172 100755
--- 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)