Explicit model load warning.

as-ideas · May 19, 2020 · 4e898a5 · 4e898a5
1 parent dea2250
commit 4e898a5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/utils/config_manager.py b/utils/config_manager.py
@@ -154,6 +154,9 @@ def load_model(self, checkpoint_path: str = None, verbose=True):
             if verbose:
                 print(f'restored weights from {checkpoint_path} at step {model.step}')
         else:
+            if manager.latest_checkpoint is None:
+                print(f'WARNING: could not find weights file. Trying to load from \n {self.weights_dir}.')
+                print('Edit data_config.yaml to point at the right log directory.')
             ckpt.restore(manager.latest_checkpoint)
             if verbose:
                 print(f'restored weights from {manager.latest_checkpoint} at step {model.step}')