Training in progress, epoch 1

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "microsoft/swinv2-base-patch4-window8-256",
+  "_name_or_path": "microsoft/swinv2-small-patch4-window8-256",
   "architectures": [
     "Swinv2ForImageClassification"
   ],
@@ -11,11 +11,11 @@
     2
   ],
   "drop_path_rate": 0.1,
-  "embed_dim": 128,
+  "embed_dim": 96,
   "encoder_stride": 32,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
-  "hidden_size": 1024,
+  "hidden_size": 768,
   "id2label": {
     "0": "HGSC",
     "1": "LGSC",
@@ -37,10 +37,10 @@
   "model_type": "swinv2",
   "num_channels": 3,
   "num_heads": [
-    4,
-    8,
-    16,
-    32
+    3,
+    6,
+    12,
+    24
   ],
   "num_layers": 4,
   "patch_size": 4,

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6792b3285deb5147396d5bfcad09f023b2918d52f450bd113f0661e4046b3129
-size 347764846
+oid sha256:75fd03c865a93a9f0e328f6fee5bc49a93ed96ae948843a090877d3c45327623
+size 196021166

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8cd1a316641b35d5e9787f71e34d5c5a6e765c749ee4e1f7b5b54c4565667a38
+size 4536