diff --git a/config/config.py b/config/config.py
index dd67082ac28bc2807ebabbccd8b9bf2739a0f37d..9673a42fe6e7bf45d60d16bfa47b378f35db4a5e 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')