mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-19 17:49:24 +01:00
9 lines
355 B
Bash
9 lines
355 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' \
|
|
--train_split train \
|
|
--val_split validation \
|
|
--use_model_config true \
|