fix image size

all_results.json

@@ -1,12 +1,12 @@
 {
     "epoch": 6.0,
-    "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_accuracy": 0.7368421052631579,
+    "eval_loss": 0.7412705421447754,
+    "eval_runtime": 0.7249,
+    "eval_samples_per_second": 183.475,
+    "eval_steps_per_second": 23.452,
+    "train_loss": 0.8289703215011442,
+    "train_runtime": 36.6127,
+    "train_samples_per_second": 169.449,
+    "train_steps_per_second": 5.408
 }

config.json

@@ -19,6 +19,7 @@
     "1": "bean_rust",
     "2": "healthy"
   },
+  "image_size": 224,
   "label2id": {
     "angular_leaf_spot": "0",
     "bean_rust": "1",

eval_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 6.0,
-    "eval_accuracy": 0.6766917293233082,
-    "eval_loss": 0.7563181519508362,
-    "eval_runtime": 0.5581,
-    "eval_samples_per_second": 238.319,
-    "eval_steps_per_second": 30.462
+    "eval_accuracy": 0.7368421052631579,
+    "eval_loss": 0.7412705421447754,
+    "eval_runtime": 0.7249,
+    "eval_samples_per_second": 183.475,
+    "eval_steps_per_second": 23.452
 }

pytorch_model.bin

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

train.py

@@ -124,6 +124,7 @@ def main():
         id2label=id2label,
         finetuning_task="image-classification",
     )
+    config.image_size = feature_extractor.size  # just a hack, sorry
 
     model = ResNetForImageClassification(config)
 
@@ -209,8 +210,6 @@ 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.8393681073429609,
-    "train_runtime": 32.431,
-    "train_samples_per_second": 191.298,
-    "train_steps_per_second": 6.105
+    "train_loss": 0.8289703215011442,
+    "train_runtime": 36.6127,
+    "train_samples_per_second": 169.449,
+    "train_steps_per_second": 5.408
 }

trainer_state.json

@@ -11,10 +11,10 @@
       "epoch": 6.0,
       "step": 198,
       "total_flos": 1064007556964352.0,
-      "train_loss": 0.8393681073429609,
-      "train_runtime": 32.431,
-      "train_samples_per_second": 191.298,
-      "train_steps_per_second": 6.105
+      "train_loss": 0.8289703215011442,
+      "train_runtime": 36.6127,
+      "train_samples_per_second": 169.449,
+      "train_steps_per_second": 5.408
     }
   ],
   "max_steps": 198,

training_args.bin

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