File size: 920 Bytes
a389d7c 67052b7 a389d7c 67052b7 a389d7c 67052b7 |
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.py \
--train_dir /trunk/shared/eebo_data/images_cropped/ \
--validation_dir /trunk/shared/eebo_data/images_cropped/ \
--output_dir ./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 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 \
--resume_from_checkpoint /trunk2/yufei/summer24/transformers/examples/pytorch/image-classification/default_model/checkpoint-67330 |