|
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_default_model/ \ |
|
--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 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 \ |
|
--resume_from_checkpoint /trunk2/yufei/summer24/transformers/examples/pytorch/image-classification/test_default_model/checkpoint-19805 |