Add cont_loss_with_uncertainty to ignore hyperparams for checkpoints

ruflab · Mar 8, 2024 · e636ef4 · e636ef4
1 parent c9eca21
commit e636ef4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/shimmer/modules/global_workspace.py b/shimmer/modules/global_workspace.py
@@ -108,6 +108,7 @@ def __init__(
                 "loss_mod",
                 "domain_descriptions",
                 "contrastive_loss",
+                "cont_loss_with_uncertainty",
                 "gw_encoders",
                 "gw_decoders",
             ]