Use 1-based indexing for epochs

Author: felker

plasma/models/builder.py

@@ -312,7 +312,9 @@ def load_model_weights(self, model, custom_path=None):
             epochs = self.get_all_saved_files()
             if len(epochs) == 0:
                 g.write_all('no previous checkpoint found\n')
-                return -1
+                # TODO(KGF): port indexing change (from "return -1") to parts
+                # of the code other than mpi_runner.py
+                return 0
             else:
                 max_epoch = max(epochs)
                 g.write_all('loading from epoch {}\n'.format(max_epoch))

plasma/models/mpi_runner.py

@@ -819,7 +819,7 @@ def mpi_train(conf, shot_list_train, shot_list_validate, loader,
     # 2019-11-06 18:27:31.698908: I ...  dynamic library libcublas.so.10
     # which are NOT suppressed by set_verbosity. See top level __init__.py
 
-    # load the latest epoch we did. Returns -1 if none exist yet
+    # load the latest epoch we did. Returns 0 if none exist yet
     e = specific_builder.load_model_weights(train_model)
     e_old = e