Training in progress, epoch 1

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "facebook/dinov2-small-imagenet1k-1-layer",
+  "_name_or_path": "facebook/dinov2-base-imagenet1k-1-layer",
   "apply_layernorm": true,
   "architectures": [
     "Dinov2ForImageClassification"
@@ -8,7 +8,7 @@
   "drop_path_rate": 0.0,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
-  "hidden_size": 384,
+  "hidden_size": 768,
   "id2label": {
     "0": "flat",
     "1": "high",
@@ -25,7 +25,7 @@
   "layerscale_value": 1.0,
   "mlp_ratio": 4,
   "model_type": "dinov2",
-  "num_attention_heads": 6,
+  "num_attention_heads": 12,
   "num_channels": 3,
   "num_hidden_layers": 12,
   "out_features": [

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fa539ec11724be633efb9b4217f75611d15a80aed8c7d45ffef68e5e845c5e27
-size 88260908
+oid sha256:e8634e00f75fb442f10411c164144d2912f3aab57ca35f33a5db41a6b224a568
+size 346366076

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ef08cd6012bab7038a507d27f75a0ad1257aa41f76cb7c01ad672054a52585b6
+oid sha256:43d90bb1387ec43eb071ac2d3e6056b9ba747e17b9118eb6005d2646647ebbe9
 size 4536