Training in progress epoch 0

all_results.json

@@ -0,0 +1 @@
+{"eval_accuracy": 0.8747300215982722, "eval_precision": 0.37475579975579976, "eval_recall": 0.2978276569176108, "eval_f1": 0.3241136083601837, "eval_specificity": 0.907816142191142, "eval_train_loss": 0.41072312716780035, "eval_epoch": 1, "eval_step": 232}

config.json

@@ -1,6 +1,6 @@
 {
-  "_name_or_path": "timm/vit_large_patch16_224.augreg_in21k_ft_in1k",
-  "architecture": "vit_large_patch16_224",
+  "_name_or_path": "timm/vit_base_patch16_224.augreg_in21k_ft_in1k",
+  "architecture": "vit_base_patch16_224",
   "architectures": [
     "TimmWrapperForImageClassification"
   ],
@@ -18,7 +18,7 @@
   ],
   "model_type": "timm_wrapper",
   "num_classes": 6,
-  "num_features": 1024,
+  "num_features": 768,
   "pretrained_cfg": {
     "classifier": "head",
     "crop_mode": "center",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e50ec5874c54e80d84918df954445388df9d4466dc0f15efd05f76d9627f98cf
-size 1213259304
+oid sha256:4ddc3bb2fc2f5f9838073789b668225fb36a08e64dd170e13ecd35335674b7aa
+size 343227200