diff --git a/PyTorch/built-in/cv/detection/RetinaNet_for_PyTorch/test/train_performance_8p.sh b/PyTorch/built-in/cv/detection/RetinaNet_for_PyTorch/test/train_performance_8p.sh index db657d20a4d58c436751a3b7c6f7031bd4e75901..90ae02a0a660748074336a204b7cca1010613810 100644 --- a/PyTorch/built-in/cv/detection/RetinaNet_for_PyTorch/test/train_performance_8p.sh +++ b/PyTorch/built-in/cv/detection/RetinaNet_for_PyTorch/test/train_performance_8p.sh @@ -111,6 +111,7 @@ fi mkdir -p $cur_path/../data ln -snf $data_path/coco $cur_path/../data/ cp train_retinanet_8p.sh $cur_path/../ +sed -i "s|total_epochs = 12|total_epochs = 1|g" $cur_path/../configs/_base_/schedules/schedule_1x.py #训练开始时间,不需要修改 start_time=$(date +%s)