fix : tile classification model

change : pywiff reader location

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')