donut-base-beans / command.txt
YufeiWeng's picture
Model save
e676bfe verified
raw
history blame
842 Bytes
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