diff --git a/edenn/models/edenn_plus.py b/edenn/models/edenn_plus.py index d28f8d5e50f98bac88986cb6be218d754e086f36..162d91c3a2a59fe91ddc2e761f725268b953d3ec 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