From: Steinar H. Gunderson <sgunderson@bigfoot.com>
Date: Fri, 9 Feb 2018 20:05:45 +0000 (+0000)
Subject: Checkpoint a bit more often.
X-Git-Url: https://git.sesse.net/?a=commitdiff_plain;h=1f69db43463b1c6a23b4c72306abd92efa5463cc;p=voxel-flow

Checkpoint a bit more often.
---

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)