diff --git a/edenn/models/edenn_plus.py b/edenn/models/edenn_plus.py index 162d91c3a2a59fe91ddc2e761f725268b953d3ec..e5be4a4f0219486bdd4a12e9e4b574e1db46d70a 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, sync_dist = True) + self.log('VAL_task_loss', t_error, prog_bar = True, sync_dist = True) # val step doesn't return anything