From 4299fbd7ff362e5ae4805e7221b6e692f6394319 Mon Sep 17 00:00:00 2001
From: bdvllrs <bdvllrs@gmail.com>
Date: Fri, 23 Feb 2024 14:34:43 +0000
Subject: [PATCH] Ignore contrastive_loss in checkpointing

---
 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 f71fe1a5..974f0744 100644
--- 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",
             ]
         )