From e636ef450599028d544f39fe465b4f5097e90b2b Mon Sep 17 00:00:00 2001
From: bdvllrs <bdvllrs@gmail.com>
Date: Fri, 8 Mar 2024 10:38:02 +0000
Subject: [PATCH] Add cont_loss_with_uncertainty to ignore hyperparams for
 checkpoints

---
 shimmer/modules/global_workspace.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/shimmer/modules/global_workspace.py b/shimmer/modules/global_workspace.py
index c17cd7b9..cf5d8dbe 100644
--- 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",
             ]