Training in progress, epoch 1

config.json

@@ -3,20 +3,20 @@
     "ResNetForImageClassification"
   ],
   "depths": [
-    2,
-    2,
-    2,
-    2
+    3,
+    4,
+    6,
+    3
   ],
   "downsample_in_bottleneck": false,
   "downsample_in_first_stage": false,
   "embedding_size": 64,
   "hidden_act": "relu",
   "hidden_sizes": [
-    64,
-    128,
     256,
-    512
+    512,
+    1024,
+    2048
   ],
   "id2label": {
     "0": "Arial",
@@ -66,7 +66,7 @@
     "Verdana Bold Italic": "20",
     "Verdana Italic": "21"
   },
-  "layer_type": "basic",
+  "layer_type": "bottleneck",
   "model_type": "resnet",
   "num_channels": 3,
   "out_features": [

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4f5a4cc46dfe779ced623af37c88cb4f9d7621ae683ad6df9b6c2d8c54fa4ab8
-size 44805376
+oid sha256:55e49546c347c3df824c4c5d202c8ae9b471c97c8b848fe5940cf9b8f06e39ed
+size 94466920

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b7146b4f092a2656ee202986f4ce798deb4f83de045fffba8640e32799c77e13
+oid sha256:f42f5f72e052fbfa6cc2668546722ddb3fc05fc174f5abf9796a4d7ead498605
 size 5713