image size

all_results.json

@@ -1,12 +1,12 @@
 {
     "epoch": 6.0,
-    "eval_accuracy": 0.6390977443609023,
-    "eval_loss": 0.7639745473861694,
-    "eval_runtime": 0.7192,
-    "eval_samples_per_second": 184.925,
-    "eval_steps_per_second": 23.637,
-    "train_loss": 0.8484223491013653,
-    "train_runtime": 32.6308,
-    "train_samples_per_second": 190.127,
-    "train_steps_per_second": 6.068
+    "eval_accuracy": 0.6766917293233082,
+    "eval_loss": 0.7563181519508362,
+    "eval_runtime": 0.5581,
+    "eval_samples_per_second": 238.319,
+    "eval_steps_per_second": 30.462,
+    "train_loss": 0.8393681073429609,
+    "train_runtime": 32.431,
+    "train_samples_per_second": 191.298,
+    "train_steps_per_second": 6.105
 }

eval_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 6.0,
-    "eval_accuracy": 0.6390977443609023,
-    "eval_loss": 0.7639745473861694,
-    "eval_runtime": 0.7192,
-    "eval_samples_per_second": 184.925,
-    "eval_steps_per_second": 23.637
+    "eval_accuracy": 0.6766917293233082,
+    "eval_loss": 0.7563181519508362,
+    "eval_runtime": 0.5581,
+    "eval_samples_per_second": 238.319,
+    "eval_steps_per_second": 30.462
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:64d031cd790aedfb724b3db0bccf4be88ab3b13655bab905ec329dfa39a8c0d6
+oid sha256:c9a849328e9031d8ed6fd38336e703681123ef5d3192b7e1fa406a348c9d120d
 size 786777

train.py

@@ -209,6 +209,8 @@ def main():
         data_collator=collate_fn,
     )
 
+    model.image_size = feature_extractor.size  # just a hack, sorry
+
     # Training
     if training_args.do_train:
         train_result = trainer.train()

train_results.json

@@ -1,7 +1,7 @@
 {
     "epoch": 6.0,
-    "train_loss": 0.8484223491013653,
-    "train_runtime": 32.6308,
-    "train_samples_per_second": 190.127,
-    "train_steps_per_second": 6.068
+    "train_loss": 0.8393681073429609,
+    "train_runtime": 32.431,
+    "train_samples_per_second": 191.298,
+    "train_steps_per_second": 6.105
 }

trainer_state.json

@@ -11,10 +11,10 @@
       "epoch": 6.0,
       "step": 198,
       "total_flos": 1064007556964352.0,
-      "train_loss": 0.8484223491013653,
-      "train_runtime": 32.6308,
-      "train_samples_per_second": 190.127,
-      "train_steps_per_second": 6.068
+      "train_loss": 0.8393681073429609,
+      "train_runtime": 32.431,
+      "train_samples_per_second": 191.298,
+      "train_steps_per_second": 6.105
     }
   ],
   "max_steps": 198,

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b5c30f97a34f52c729b4381ca5704c7c50d318011f10167ec3bfaa9db9be0777
+oid sha256:dc0fca983cae4cdd3fca2d5ada53febd07b7afe88364d8f80a3b5bb9e708ed1f
 size 3247