Training in progress, epoch 0

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "microsoft/swin-tiny-patch4-window7-224",
+  "_name_or_path": "microsoft/swin-base-patch4-window7-224-in22k",
   "architectures": [
     "SwinForImageClassification"
   ],
@@ -7,15 +7,15 @@
   "depths": [
     2,
     2,
-    6,
+    18,
     2
   ],
   "drop_path_rate": 0.1,
-  "embed_dim": 96,
+  "embed_dim": 128,
   "encoder_stride": 32,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
-  "hidden_size": 768,
+  "hidden_size": 1024,
   "image_size": 224,
   "initializer_range": 0.02,
   "layer_norm_eps": 1e-05,
@@ -23,10 +23,10 @@
   "model_type": "swin",
   "num_channels": 3,
   "num_heads": [
-    3,
-    6,
-    12,
-    24
+    4,
+    8,
+    16,
+    32
   ],
   "num_layers": 4,
   "out_features": [

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f8f4a6479e9142338f8d2d1d75a372868b614d94f9d015b4efe57af542083dd9
-size 110342832
+oid sha256:06baf302f8ba8649c5634e134257896b7c3f996dd4294856162ef81fa5a8929d
+size 347498816

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:91dbb7171b585ea1a43e4781a15c369115f5af76ff48d2dd794f5ee2a3191171
+oid sha256:07130d200f60285e926ce3ec6d8808f55f5bb0127b694dc40495f36a39a48858
 size 4219