fix reload.

mygit007hub · Aug 11, 2017 · 011d240 · 011d240
1 parent 4b6e7ce
commit 011d240
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/model/train_val.py b/lib/model/train_val.py
@@ -203,8 +203,8 @@ def initialize(self, sess):
 
   def restore(self, sess, sfile, nfile):
     # Get the most recent snapshot and restore
-    np_paths = [sfile]
-    ss_paths = [nfile]
+    np_paths = [nfile]
+    ss_paths = [sfile]
     # Restore model from snapshots
     last_snapshot_iter = self.from_snapshot(sess, sfile, nfile)
     # Set the learning rate