diff --git a/barlow_twin_like/main.py b/barlow_twin_like/main.py index 2542c308a697f554e67e36a3e4a8631222fb67ba..01effe173511771c7f8650c1583e1892be2cd407 100644 --- a/barlow_twin_like/main.py +++ b/barlow_twin_like/main.py @@ -61,7 +61,7 @@ def test_representation(model, data_val, epoch, wandb): print('Val epoch {}, loss : {:.3f}'.format(epoch, losses)) if wandb is not None: - wdb.log({"train loss": losses, 'train epoch': epoch}) + wdb.log({"val loss": losses, 'val epoch': epoch}) return losses