From 3b2a5d7f3db10bfe6c734513c96c87958f5ca99a Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 20 May 2025 15:18:26 +0200
Subject: [PATCH] fix : wandb log barlow
---
barlow_twin_like/main.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/barlow_twin_like/main.py b/barlow_twin_like/main.py
index 2542c30..01effe1 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
--
GitLab