multi_gpu_train.sh 173 Bytes
Newer Older
wangsen's avatar
wangsen committed
1
2
# export NCCL_P2P_DISABLE=1
CUDA_VISIBLE_DEVICES=0,1,2,3 python3 -m paddle.distributed.launch tools/train.py --config_file "config/icdar2015_resnet50_FPN_DBhead_polyLR.yaml"