Fix issue by initialising opt_start_epoch as None

vue1999 · Jan 21, 2025 · 5da7dff · 5da7dff
1 parent f2243bc
commit 5da7dff
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mace/cli/run_train.py b/mace/cli/run_train.py
@@ -607,6 +607,7 @@ def run(args: argparse.Namespace) -> None:
 
     start_epoch = 0
     restart_lbfgs = False
+    opt_start_epoch = None
     if args.restart_latest:
         try:
             opt_start_epoch = checkpoint_handler.load_latest(