12 lines
318 B
Bash
12 lines
318 B
Bash
export CUDA_VISIBLE_DEVICES=0,1,2
|
|
python -m torch.distributed.launch \
|
|
--nproc_per_node=3 train.py \
|
|
--respath checkpoints/train_STDC2-Seg/ \
|
|
--backbone STDCNet1446 \
|
|
--mode train \
|
|
--n_workers_train 12 \
|
|
--n_workers_val 1 \
|
|
--max_iter 60000 \
|
|
--use_boundary_8 True \
|
|
--pretrain_path checkpoints/STDCNet1446_76.47.tar
|