diff --git a/PyTorch/built-in/docs/dist_train_demo/README.md b/PyTorch/built-in/docs/dist_train_demo/README.md index 45cc3d80086c56d266684349dbfa09f4fe9f5c1b..58b40cc1608c54d94f970faffac6cc355c0bd5c3 100644 --- a/PyTorch/built-in/docs/dist_train_demo/README.md +++ b/PyTorch/built-in/docs/dist_train_demo/README.md @@ -11,7 +11,7 @@ ## 简述 -基于示例模型脚本,针对shell脚本启动、python启动、mp.spawn启动、torchrun启动、torch_npu_run启动五种方式,分别提供单机八卡和双机十六卡端到端训练示例脚本。 +基于示例模型脚本,针对shell脚本启动、torch.distributed.launch启动、mp.spawn启动、torchrun启动、torch_npu_run启动五种方式,分别提供单机八卡和双机十六卡端到端训练示例脚本。 # 准备训练环境 @@ -62,7 +62,7 @@ bash start_train_16p_slave_addr.sh # 从节点 ``` - - python脚本启动。 + - torch.distributed.launch脚本启动。 启动单机八卡训练。 diff --git a/PyTorch/built-in/docs/dist_train_demo/train_16p_python.py b/PyTorch/built-in/docs/dist_train_demo/train_16p_torch.distributed.launch.py similarity index 100% rename from PyTorch/built-in/docs/dist_train_demo/train_16p_python.py rename to PyTorch/built-in/docs/dist_train_demo/train_16p_torch.distributed.launch.py diff --git a/PyTorch/built-in/docs/dist_train_demo/train_8p_python.py b/PyTorch/built-in/docs/dist_train_demo/train_8p_torch.distributed.launch.py similarity index 100% rename from PyTorch/built-in/docs/dist_train_demo/train_8p_python.py rename to PyTorch/built-in/docs/dist_train_demo/train_8p_torch.distributed.launch.py diff --git a/PyTorch/dev/cv/image_classification/SlowFast_ID0646_for_PyTorch/detectron2/projects/Panoptic-DeepLab/configs/COCO-PanopticSegmentation/panoptic_deeplab_R_52_os16_mg124_poly_200k_bs64_crop_640_640_coco_dsconv.yaml b/PyTorch/dev/cv/image_classification/SlowFast_ID0646_for_PyTorch/detectron2/projects/Panoptic-DeepLab/configs/COCO-PanopticSegmentation/panoptic_deeplab_R_52_os16_mg124_poly_200k_bs64_crop_640_640_coco_dsconv.yaml deleted file mode 100644 index 74f448ddba0a2cb8de27ca0b8f943d5cc6471978..0000000000000000000000000000000000000000 --- a/PyTorch/dev/cv/image_classification/SlowFast_ID0646_for_PyTorch/detectron2/projects/Panoptic-DeepLab/configs/COCO-PanopticSegmentation/panoptic_deeplab_R_52_os16_mg124_poly_200k_bs64_crop_640_640_coco_dsconv.yaml +++ /dev/null @@ -1,42 +0,0 @@ -_BASE_: ../Cityscapes-PanopticSegmentation/Base-PanopticDeepLab-OS16.yaml -MODEL: - WEIGHTS: "detectron2://DeepLab/R-52.pkl" - PIXEL_MEAN: [123.675, 116.280, 103.530] - PIXEL_STD: [58.395, 57.120, 57.375] - BACKBONE: - NAME: "build_resnet_deeplab_backbone" - RESNETS: - DEPTH: 50 - NORM: "SyncBN" - RES5_MULTI_GRID: [1, 2, 4] - STEM_TYPE: "deeplab" - STEM_OUT_CHANNELS: 128 - STRIDE_IN_1X1: False - SEM_SEG_HEAD: - NUM_CLASSES: 133 - LOSS_TOP_K: 1.0 - USE_DEPTHWISE_SEPARABLE_CONV: True - PANOPTIC_DEEPLAB: - STUFF_AREA: 4096 - NMS_KERNEL: 41 - SIZE_DIVISIBILITY: 640 - USE_DEPTHWISE_SEPARABLE_CONV: True -DATASETS: - TRAIN: ("coco_2017_train_panoptic",) - TEST: ("coco_2017_val_panoptic",) -SOLVER: - BASE_LR: 0.0005 - MAX_ITER: 200000 - IMS_PER_BATCH: 64 -INPUT: - FORMAT: "RGB" - GAUSSIAN_SIGMA: 8 - MIN_SIZE_TRAIN: !!python/object/apply:eval ["[int(x * 0.1 * 640) for x in range(5, 16)]"] - MIN_SIZE_TRAIN_SAMPLING: "choice" - MIN_SIZE_TEST: 640 - MAX_SIZE_TRAIN: 960 - MAX_SIZE_TEST: 640 - CROP: - ENABLED: True - TYPE: "absolute" - SIZE: (640, 640) \ No newline at end of file