Checkpoint before migrating examples

tensorpack · Oct 21, 2017 · 9268bc8 · 9268bc8
1 parent 0be707e
commit 9268bc8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/tensorpack/libinfo.py b/tensorpack/libinfo.py
@@ -33,4 +33,4 @@
     _HAS_TF = False
 
 
-__version__ = '0.6.0'
+__version__ = '0.6.1'
diff --git a/tensorpack/trainv2/trainers.py b/tensorpack/trainv2/trainers.py
@@ -178,3 +178,7 @@ def initialize(self, session_creator, session_init):
                 "To use a custom session config, pass it to tf.train.Server.")
         super(DistributedTrainerReplicated, self).initialize(
             get_distributed_session_creator(), session_init)
+
+    @property
+    def _main_tower_vs_name(self):
+        return "tower0"