Skip to content
Snippets Groups Projects
Commit 3b2a5d7f authored by Schneider Leo's avatar Schneider Leo
Browse files

fix : wandb log barlow

parent 470dba92
No related branches found
No related tags found
No related merge requests found
...@@ -61,7 +61,7 @@ def test_representation(model, data_val, epoch, wandb): ...@@ -61,7 +61,7 @@ def test_representation(model, data_val, epoch, wandb):
print('Val epoch {}, loss : {:.3f}'.format(epoch, losses)) print('Val epoch {}, loss : {:.3f}'.format(epoch, losses))
if wandb is not None: if wandb is not None:
wdb.log({"train loss": losses, 'train epoch': epoch}) wdb.log({"val loss": losses, 'val epoch': epoch})
return losses return losses
......
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