Remove unused losses key in checkpoint

cvg · Oct 23, 2023 · 28450b5 · 28450b5
1 parent e0104fd
commit 28450b5
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 6 deletions.
diff --git a/gluefactory/train.py b/gluefactory/train.py
@@ -341,7 +341,6 @@ def sigint_handler(signal, frame):
         logger.info(
             "Starting training with configuration:\n%s", OmegaConf.to_yaml(conf)
         )
-    losses_ = None
 
     def trace_handler(p):
         # torch.profiler.tensorboard_trace_handler(str(output_dir))
@@ -551,7 +550,6 @@ def trace_handler(p):
                             optimizer,
                             lr_scheduler,
                             conf,
-                            losses_,
                             results,
                             best_eval,
                             epoch,
@@ -586,7 +584,6 @@ def trace_handler(p):
                     optimizer,
                     lr_scheduler,
                     conf,
-                    losses_,
                     results,
                     best_eval,
                     epoch,
@@ -605,7 +602,6 @@ def trace_handler(p):
                 optimizer,
                 lr_scheduler,
                 conf,
-                losses_,
                 results,
                 best_eval,
                 epoch,

diff --git a/gluefactory/utils/experiments.py b/gluefactory/utils/experiments.py
@@ -97,7 +97,6 @@ def save_experiment(
     optimizer,
     lr_scheduler,
     conf,
-    losses,
     results,
     best_eval,
     epoch,
@@ -116,7 +115,6 @@ def save_experiment(
         "lr_scheduler": lr_scheduler.state_dict(),
         "conf": OmegaConf.to_container(conf, resolve=True),
         "epoch": epoch,
-        "losses": losses,
         "eval": results,
     }
     if cp_name is None: