Ignore contrastive_loss in checkpointing

ruflab · Feb 23, 2024 · 4299fbd · 4299fbd
1 parent 2ddda1f
commit 4299fbd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/shimmer/modules/global_workspace.py b/shimmer/modules/global_workspace.py
@@ -56,6 +56,7 @@ def __init__(
                 "loss_mod",
                 "domain_descriptions",
                 "coef_buffers",
+                "contrastive_loss",
             ]
         )