Training in progress, epoch 1

config.json

@@ -1,21 +1,13 @@
 {
-  "_name_or_path": "diegokauer/int-pet-classifier",
+  "_name_or_path": "google/vit-base-patch16-224-in21k",
   "architectures": [
-    "Swinv2ForImageClassification"
+    "ViTForImageClassification"
   ],
   "attention_probs_dropout_prob": 0.0,
-  "depths": [
-    2,
-    2,
-    18,
-    2
-  ],
-  "drop_path_rate": 0.1,
-  "embed_dim": 128,
-  "encoder_stride": 32,
+  "encoder_stride": 16,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
-  "hidden_size": 1024,
+  "hidden_size": 768,
   "id2label": {
     "0": "No Reportado",
     "1": "Cirug\u00eda",
@@ -25,8 +17,9 @@
     "5": "Estudio Adicional",
     "6": "Otro"
   },
-  "image_size": 256,
+  "image_size": 224,
   "initializer_range": 0.02,
+  "intermediate_size": 3072,
   "label2id": {
     "Cirug\u00eda": 1,
     "Cuidados Paliativos": 4,
@@ -36,29 +29,14 @@
     "QMT": 2,
     "RadioTerapia": 3
   },
-  "layer_norm_eps": 1e-05,
-  "mlp_ratio": 4.0,
-  "model_type": "swinv2",
+  "layer_norm_eps": 1e-12,
+  "model_type": "vit",
+  "num_attention_heads": 12,
   "num_channels": 3,
-  "num_heads": [
-    4,
-    8,
-    16,
-    32
-  ],
-  "num_layers": 4,
-  "patch_size": 4,
-  "path_norm": true,
-  "pretrained_window_sizes": [
-    0,
-    0,
-    0,
-    0
-  ],
+  "num_hidden_layers": 12,
+  "patch_size": 16,
   "problem_type": "multi_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.35.2",
-  "use_absolute_embeddings": false,
-  "window_size": 8
+  "transformers_version": "4.35.2"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f83af8f94d7a1804c3d85af7e306b7b5cd5695af8a1aba98c1a64730cfb070c4
-size 347665996
+oid sha256:10545eb5b3c13967855b0b88dc77ae48039af686c5fe941e5249ecc684501bc1
+size 343239356

preprocessor_config.json

@@ -2,22 +2,21 @@
   "do_normalize": true,
   "do_rescale": true,
   "do_resize": true,
-  "feature_extractor_type": "ViTFeatureExtractor",
   "image_mean": [
-    0.485,
-    0.456,
-    0.406
+    0.5,
+    0.5,
+    0.5
   ],
   "image_processor_type": "ViTImageProcessor",
   "image_std": [
-    0.229,
-    0.224,
-    0.225
+    0.5,
+    0.5,
+    0.5
   ],
-  "resample": 3,
+  "resample": 2,
   "rescale_factor": 0.00392156862745098,
   "size": {
-    "height": 256,
-    "width": 256
+    "height": 224,
+    "width": 224
   }
 }

runs/Jan03_18-19-14_c96e1e5ad27c/events.out.tfevents.1704305956.c96e1e5ad27c.85687.1

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0f6be0ca90aa105407e3afea4cb2dc65e024e89605eac215df419e717af8784f
-size 42562
+oid sha256:622fc8f59e61f09c910ca2e02810aa60d8ee3751e26f7e11a3090387a82c70db
+size 55036

runs/Jan03_19-03-51_c96e1e5ad27c/events.out.tfevents.1704308634.c96e1e5ad27c.98413.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:453ee83301cdf83fef9920c60d35728d1f359d7627b9753c671767298d05e6a6
+size 11398

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:696a14ed356599b7c9bb04085c84ec6439db007ca9dd7f9a17625d22fc8b8ffd
+oid sha256:6cb200dc6e1d52e8be1ae8af7838d39fde0a48d1a31068718d51861c7ea0964b
 size 4600