run_training.sh 128 Bytes
Newer Older
20220418012 committed
1 2 3 4 5
#!/usr/bin/env bash
python3 -m torch.distributed.launch --nproc_per_node=3  \
train.py \
--config config.json \
--gpu '2,5,6' \