Skip to content
Snippets Groups Projects

still fixing checkpointing

parent ec1aec35
No related branches found
No related tags found
No related merge requests found
...@@ -415,7 +415,7 @@ class Udenn_npl(pl.LightningModule): ...@@ -415,7 +415,7 @@ class Udenn_npl(pl.LightningModule):
t_error = self.loss(out, flow) # task loss 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 # val step doesn't return anything
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment