From 2085fd0e1e77565861c6a200fd7733c0c0f854a1 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 20 May 2025 16:12:18 +0200
Subject: [PATCH] fix : wandb log barlow
---
barlow_twin_like/config.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/barlow_twin_like/config.py b/barlow_twin_like/config.py
index ee4a58b..a9aeb92 100644
--- a/barlow_twin_like/config.py
+++ b/barlow_twin_like/config.py
@@ -13,7 +13,7 @@ def load_args_barlow():
parser.add_argument('--lambd', type=float, default=0.005)
parser.add_argument('--opti', type=str, default='adam')
parser.add_argument('--model', type=str, default='ResNet18')
- parser.add_argument('--projector', type=str, default='256-128-64')
+ parser.add_argument('--projector', type=str, default='2048-4096')
parser.add_argument('--sampler', type=str, default=None)
parser.add_argument('--dataset_train_dir', type=str, default='data/processed_data_wiff_clean_005_10000_apex/npy_image/train data')
parser.add_argument('--dataset_val_dir', type=str, default='data/processed_data_wiff_clean_005_10000_apex/npy_image/val data')
--
GitLab