update medical

bubbliiiing · Apr 17, 2022 · 016eaee · 016eaee
1 parent 216fa5b
commit 016eaee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train_medical.py b/train_medical.py
@@ -53,10 +53,10 @@
     #                   Windows系统下默认使用DP模式调用所有显卡，不支持DDP。
     #   DP模式：
     #       设置            distributed = False
-    #       在终端中输入    CUDA_VISIBLE_DEVICES=0,1 python train.py
+    #       在终端中输入    CUDA_VISIBLE_DEVICES=0,1 python train_medical.py
     #   DDP模式：
     #       设置            distributed = True
-    #       在终端中输入    CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 train.py
+    #       在终端中输入    CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 train_medical.py
     #---------------------------------------------------------------------#
     distributed     = False
     #---------------------------------------------------------------------#