train.sh 570 Bytes
Newer Older
chenzk's avatar
v1.0  
chenzk committed
1
2
3
python -m torch.distributed.launch --nproc_per_node=4 --master_port 12312 --use_env main.py --model repvit_m0_9 --data-path ./cifar100 --data-set CIFAR --epochs 300 --dist-eval --distillation-type none
# python -m torch.distributed.launch --nproc_per_node=4 --master_port 12346 --use_env main.py --model repvit_m0_9 --data-path ./imagenet --teacher-path regnety_160-a5fe301d.pth --epochs 1 --dist-eval
# NCCL_P2P_DISABLE=1 python -m torch.distributed.launch --nproc_per_node=8 --master_port 12346 --use_env main.py --model repvit_m0_9 --data-path ~/imagenet --dist-eval