File size: 842 Bytes
e466b8e b64eba1 e466b8e e676bfe | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | 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 |