From ea8967e3361a18fee3b15cf4ac8fc1497d69bd0f Mon Sep 17 00:00:00 2001
From: ah02299 <ah02299@surrey.ac.uk>
Date: Fri, 16 Feb 2024 12:08:38 +0000
Subject: [PATCH] still fixing checkpointing
---
edenn/models/edenn_plus.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/edenn/models/edenn_plus.py b/edenn/models/edenn_plus.py
index d28f8d5..162d91c 100644
--- a/edenn/models/edenn_plus.py
+++ b/edenn/models/edenn_plus.py
@@ -415,7 +415,7 @@ class Udenn_npl(pl.LightningModule):
t_error = self.loss(out, flow) # task loss
- self.log('VAL_task_loss', t_error)
+ self.log('VAL_task_loss', t_error, sync_dist = True)
# val step doesn't return anything
--
GitLab