yjh0410 пре 2 година
родитељ
комит
d5dcd3f0ff
1 измењених фајлова са 1 додато и 1 уклоњено
  1. 1 1
      train_ddp.sh

+ 1 - 1
train_ddp.sh

@@ -16,4 +16,4 @@ python -m torch.distributed.run --nproc_per_node=8 train.py \
                                                     --fp16 \
                                                     --sybn \
                                                     --multi_scale \
-                                                    # --resume weights/coco/yolovx_s/yolox_s_best.pth \
+                                                    --resume weights/coco/yolovx_s/yolox_s_best.pth \