12 lines
317 B
Bash
12 lines
317 B
Bash
export CUDA_VISIBLE_DEVICES=0,1,2
|
|
python -m torch.distributed.launch \
|
|
--nproc_per_node=3 train.py \
|
|
--respath checkpoints/train_STDC1-Seg/ \
|
|
--backbone STDCNet813 \
|
|
--mode train \
|
|
--n_workers_train 12 \
|
|
--n_workers_val 1 \
|
|
--max_iter 60000 \
|
|
--use_boundary_8 True \
|
|
--pretrain_path checkpoints/STDCNet813M_73.91.tar
|