update

train.sh

@@ -83,7 +83,8 @@ if [ $WORLD_SIZE == 1 ]; then
             --fp16 \
             --multi_scale
 elif [[ $WORLD_SIZE -gt 1 && $WORLD_SIZE -le 8 ]]; then
-    python -m torch.distributed.run --nproc_per_node=${WORLD_SIZE} --master_port 1669 train.py \
+    MASTER_PORT=$7
+    python -m torch.distributed.run --nproc_per_node=${WORLD_SIZE} --master_port ${MASTER_PORT} train.py \
             --cuda \
             -dist \
             --dataset ${DATASET} \