| python run_image_classification.py \ | |
| --train_dir /trunk/shared/eebo_data/images_cropped/ \ | |
| --validation_dir /trunk/shared/eebo_data/images_cropped/ \ | |
| --output_dir ./microsoft_beit-base-patch16-224-pt22k-ft22k/ \ | |
| --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 10 \ | |
| --per_device_train_batch_size 64 \ | |
| --per_device_eval_batch_size 64 \ | |
| --logging_strategy steps \ | |
| --logging_steps 10 \ | |
| --eval_strategy epoch \ | |
| --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/beit-base-patch16-224-pt22k-ft22k \ | |
| --resume_from_checkpoint /trunk2/yufei/summer24/transformers/examples/pytorch/image-classification/microsoft_beit-base-patch16-224-pt22k-ft22k/checkpoint-33670 |