Files
modelscope/examples/pytorch/image_classification/run_train.sh

6 lines
268 B
Bash

PYTHONPATH=. python -m torch.distributed.launch --nproc_per_node=2 \
examples/pytorch/image_classification/finetune_image_classification.py \
--num_classes 2 \
--train_dataset_name 'tany0699/cats_and_dogs' \
--val_dataset_name 'tany0699/cats_and_dogs'