From 1f69db43463b1c6a23b4c72306abd92efa5463cc Mon Sep 17 00:00:00 2001
From: "Steinar H. Gunderson" <sgunderson@bigfoot.com>
Date: Fri, 9 Feb 2018 20:05:45 +0000
Subject: [PATCH] Checkpoint a bit more often.

---
 voxel_flow_train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)
 
-- 
2.39.2