|
- 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
|