From: Steinar H. Gunderson <sgunderson@bigfoot.com>
Date: Fri, 9 Feb 2018 20:05:17 +0000 (+0000)
Subject: Fix loading from checkpoints again.
X-Git-Url: https://git.sesse.net/?p=voxel-flow;a=commitdiff_plain;h=bb5e196c9163ea49124ef98725c6fac928c67f4b

Fix loading from checkpoints again.
---

diff --git a/voxel_flow_train.py b/voxel_flow_train.py
index 42f91af..7d39968 100755
--- a/voxel_flow_train.py
+++ b/voxel_flow_train.py
@@ -115,6 +115,7 @@ def train(dataset_frame1, dataset_frame2, dataset_frame3):
       restorer.restore(sess, ckpt.model_checkpoint_path)
       print('%s: Pre-trained model restored from %s' %
         (datetime.now(), ckpt.model_checkpoint_path))
+      sess.run([batch_frame1.initializer, batch_frame2.initializer, batch_frame3.initializer])
     else:
       # Build an initialization operation to run below.
       init = tf.initialize_all_variables()