diff --git a/PyTorch/dev/cv/image_classification/FasterRCNN_ID0100_for_PyTorch/test/train_full_1p.sh b/PyTorch/dev/cv/image_classification/FasterRCNN_ID0100_for_PyTorch/test/train_full_1p.sh index 5fdf57f605cc54544ba5c75c6b8629fd09d22f44..fffcbb874eec56ee3a9da54bdd96a7bd7ca6a6bf 100644 --- a/PyTorch/dev/cv/image_classification/FasterRCNN_ID0100_for_PyTorch/test/train_full_1p.sh +++ b/PyTorch/dev/cv/image_classification/FasterRCNN_ID0100_for_PyTorch/test/train_full_1p.sh @@ -1,5 +1,7 @@ #!/bin/bash - +export HCCL_WHITELIST_DISABLE=1 +export MASTER_ADDR=127.0.0.1 +export MASTER_PORT=23456 #当前路径,不需要修改 cur_path=`pwd` #export ASCEND_SLOG_PRINT_TO_STDOUT=0 @@ -96,7 +98,7 @@ sed -i "s|\"coco_2017_train\": (\"coco/train2017\", \"coco/annotations/instances sed -i "s|\"coco_2017_val\": (\"coco/val2017\", \"coco/annotations/instances_val2017.json\")|\"coco_2017_val\": (\"$data_path/coco/val2017\", \"$data_path/coco/annotations/instances_val2017.json\")|g" $cur_path/../detectron2/data/datasets/builtin.py sed -i "s|WEIGHTS: \"detectron2://ImageNetPretrained/MSRA/R-101.pkl\"|WEIGHTS: \"$data_path/R-101.pkl\"|g" $cur_path/../configs/COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml -python3 setup.py build develop > $cur_path/../log.txt +#python3 setup.py build develop > $cur_path/../log.txt #训练开始时间,不需要修改 start_time=$(date +%s)