Model save

README.md

@@ -1,8 +1,6 @@
 ---
 base_model: microsoft/dit-base-finetuned-rvlcdip
 tags:
-- image-classification
-- vision
 - generated_from_trainer
 metrics:
 - f1
@@ -18,8 +16,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [microsoft/dit-base-finetuned-rvlcdip](https://huggingface.co/microsoft/dit-base-finetuned-rvlcdip) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0385
-- F1: 0.9558
+- F1: 0.9705
+- Loss: 0.0391
 
 ## Model description
 

command

@@ -1 +1 @@
-python run_image_classification_val_test.py     --train_dir /trunk/shared/eebo_data/images_cropped/     --validation_dir /trunk/shared/eebo_data/images_cropped/     --output_dir ./test_microsoft_dit/     --remove_unused_columns False     --label_column_name labels     --do_train     --do_eval     --push_to_hub     --push_to_hub_model_id donut-base-beans     --learning_rate 3e-5     --num_train_epochs 5     --per_device_train_batch_size 32     --per_device_eval_batch_size 32     --logging_strategy steps     --logging_steps 10     --eval_strategy step     --save_strategy epoch     --load_best_model_at_end True     --save_total_limit 5     --seed 1337     --ignore_mismatched_sizes     --image_column_name image_url     --label_column_name label     --model_name_or_path microsoft/dit-base-finetuned-rvlcdip     --resume_from_checkpoint /trunk2/yufei/summer24/transformers/examples/pytorch/image-classification/test_microsoft_dit/checkpoint-9905
+python run_image_classification_val_test.py     --train_dir /trunk/shared/eebo_data/images_cropped/     --validation_dir /trunk/shared/eebo_data/images_cropped/     --output_dir ./test_microsoft_dit/     --remove_unused_columns False     --label_column_name labels     --do_train     --do_eval     --push_to_hub     --push_to_hub_model_id donut-base-beans     --learning_rate 3e-5     --num_train_epochs 5     --per_device_train_batch_size 32     --per_device_eval_batch_size 32     --logging_strategy steps     --logging_steps 10     --eval_strategy steps     --save_strategy steps     --load_best_model_at_end True     --save_total_limit 5     --seed 1337     --ignore_mismatched_sizes     --image_column_name image_url     --label_column_name label     --model_name_or_path microsoft/dit-base-finetuned-rvlcdip     --resume_from_checkpoint /trunk2/yufei/summer24/transformers/examples/pytorch/image-classification/test_microsoft_dit/checkpoint-9905

config.json

@@ -48,7 +48,6 @@
     3,
     6
   ],
-  "problem_type": "regression",
   "reshape_hidden_states": true,
   "semantic_loss_ignore_index": 255,
   "stage_names": [

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:00ae7df97b5917b24e23b5346e8492a354e57422b7b4226410beed0d77d0839f
-size 5368
+oid sha256:e361d94f5aae640777c6cdb9ee0b26da6a62d380a3516cf93bf2b20be36301bd
+size 5304