From 0b1f4c7e540f378a91da6a07dad5a20fede035e7 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 18 Jun 2025 16:46:01 +0200
Subject: [PATCH] fix : tile classification model change : pywiff reader
location
---
config/config.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/config/config.py b/config/config.py
index dd67082..9673a42 100644
--- a/config/config.py
+++ b/config/config.py
@@ -10,7 +10,7 @@ def load_args():
parser.add_argument('--noise_threshold', type=int, default=0)
parser.add_argument('--lr', type=float, default=0.001)
parser.add_argument('--batch_size', type=int, default=8)
- parser.add_argument('--model', type=str, default='ResNet50')
+ parser.add_argument('--model', type=str, default='ResNet18')
parser.add_argument('--model_type', type=str, default='duo')
parser.add_argument('--dataset_train_dir', type=str, default='data/processed_data_wiff/npy_image/train_data')
parser.add_argument('--dataset_val_dir', type=str, default='data/processed_data_wiff/npy_image/test_data')
--
GitLab