mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-23 11:39:26 +01:00
82482b3e96
Based on feat/0131/nlp_args branch, the original code review: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11408570 Support for running finetuning from the command line with training args, Compatible with the configuration optimization.
6 lines
247 B
Bash
6 lines
247 B
Bash
PYTHONPATH=. python -m torch.distributed.launch --nproc_per_node=2 \
|
|
examples/pytorch/finetune_image_classification.py \
|
|
--num_classes 2 \
|
|
--train_dataset_name 'tany0699/cats_and_dogs' \
|
|
--val_dataset_name 'tany0699/cats_and_dogs'
|