fix minor error during checkpoint loading

april-tools · Mar 15, 2024 · 50ae243 · 50ae243
1 parent e7f846f
commit 50ae243
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/scripts/engine.py b/src/scripts/engine.py
@@ -295,8 +295,9 @@ def run(self):
             # If alternate checkpoint hparams given, replace values in hparams from CL
             checkpoint_args = copy(self.args)
             for hp in self.args.checkpoint_hparams.split(';'):
-                hp_name, hp_value = hp.split('=')
-                checkpoint_args.__setattr__(hp_name.replace('-', '_'), hp_value)
+                if hp:
+                    hp_name, hp_value = hp.split('=')
+                    checkpoint_args.__setattr__(hp_name.replace('-', '_'), hp_value)
 
             checkpoint_run_id = build_run_id(checkpoint_args)
             checkpoint_exp_path = setup_experiment_path(