Training in progress, step 30

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "microsoft/swin-large-patch4-window12-384-in22k",
+  "_name_or_path": "microsoft/swin-tiny-patch4-window7-224",
   "architectures": [
     "SwinForImageClassification"
   ],
@@ -7,15 +7,15 @@
   "depths": [
     2,
     2,
-    18,
+    6,
     2
   ],
   "drop_path_rate": 0.1,
-  "embed_dim": 192,
+  "embed_dim": 96,
   "encoder_stride": 32,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
-  "hidden_size": 1536,
+  "hidden_size": 768,
   "id2label": {
     "0": "Ron00ml",
     "1": "Ron01ml",
@@ -24,7 +24,7 @@
     "4": "Ron04ml",
     "5": "Ron05ml"
   },
-  "image_size": 384,
+  "image_size": 224,
   "initializer_range": 0.02,
   "label2id": {
     "Ron00ml": "0",
@@ -39,10 +39,10 @@
   "model_type": "swin",
   "num_channels": 3,
   "num_heads": [
+    3,
     6,
     12,
-    24,
-    48
+    24
   ],
   "num_layers": 4,
   "out_features": [
@@ -65,5 +65,5 @@
   "torch_dtype": "float32",
   "transformers_version": "4.48.3",
   "use_absolute_embeddings": false,
-  "window_size": 12
+  "window_size": 7
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7b55fa4094c8b61e594e165c0665e5d0f269bdf7a86b5e01c304169ddb57c43f
-size 784869336
+oid sha256:c605474959df86eec0046894f77e7215dae288b17dce0a8f339fe44267421a4a
+size 110355136

preprocessor_config.json

@@ -17,7 +17,7 @@
   "resample": 3,
   "rescale_factor": 0.00392156862745098,
   "size": {
-    "height": 384,
-    "width": 384
+    "height": 224,
+    "width": 224
   }
 }

runs/Feb21_17-49-19_cb75bcdc2e15/events.out.tfevents.1740160173.cb75bcdc2e15.530.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:94523afd18432acbb8839eafdd354c49941a9f5e1c93bfa1d9b51097540aea67
+size 6575

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d32371f7766ddb5ca9318b3156a7da91b8c27c69e79a79cb7cfd7d9505c6f329
+oid sha256:0bf0d412926a919986fa0ba5fecefe5509b4a33991405074b0dbd8b82714d1a4
 size 5368