From a7c21414ca3b7a27a25543601f03e5e8f20819f6 Mon Sep 17 00:00:00 2001 From: "hongliang.yuan" Date: Tue, 21 Oct 2025 11:19:53 +0800 Subject: [PATCH] update execute script new model path --- tests/executables/bert/init_torch.sh | 2 +- .../bert/train_bert_default_amp_dist_1x8_torch.sh | 2 +- .../bert/train_bert_pretraining_amp_dist_1x8_tf.sh | 2 +- .../train_bert_pretraining_amp_dist_1x8_torch.sh | 4 ++-- tests/executables/conformer/init_torch.sh | 2 +- .../train_conformer_librispeech_dist_1x8_torch.sh | 2 +- tests/executables/dali/train_resnet50_dali_torch.sh | 2 +- .../train_efficientb4_imagenet_amp_dist_torch.sh | 2 +- tests/executables/fairmot/init_torch.sh | 2 +- .../fairmot/train_fairmot_hrnet32_dist_torch.sh | 2 +- tests/executables/maskrcnn/init_torch.sh | 2 +- .../maskrcnn/train_maskrcnn_resnet50_amp_torch.sh | 2 +- tests/executables/mobilenetv3/init_torch.sh | 2 +- .../mobilenetv3/train_mobilenetv3_large_amp_torch.sh | 2 +- tests/executables/resnet/init_paddle.sh | 2 +- tests/executables/resnet/train_resnet50_amp_torch.sh | 2 +- .../executables/resnet/train_resnet50_dist_paddle.sh | 2 +- tests/executables/resnet/train_resnet50_dist_tf.sh | 2 +- .../retinanet/train_retinanet_amp_torch.sh | 2 +- tests/executables/ssd/init_tf.sh | 2 +- tests/executables/ssd/init_torch.sh | 2 +- tests/executables/ssd/train_ssd_amp_tf.sh | 2 +- tests/executables/ssd/train_ssd_amp_torch.sh | 2 +- tests/executables/stable-diffusion/init_torch.sh | 2 +- .../train_sd2.1_pokemon_dist_1x8_torch.sh | 2 +- .../train_unet3d_kits19_stage3_dist_1x8_torch.sh | 2 +- tests/executables/yolov5/init_torch.sh | 2 +- .../yolov5/train_yolov5s_coco_amp_torch.sh | 2 +- tests/run_test.py | 12 ++++++------ 29 files changed, 35 insertions(+), 35 deletions(-) diff --git a/tests/executables/bert/init_torch.sh b/tests/executables/bert/init_torch.sh index bc93aa7fd..5280ab41f 100644 --- a/tests/executables/bert/init_torch.sh +++ b/tests/executables/bert/init_torch.sh @@ -58,7 +58,7 @@ fi # install sdk -cd ${CURRENT_DIR}/../../nlp/language_model/bert_sample/pytorch/base +cd ${CURRENT_DIR}/../../models/nlp/language_model/bert_sample/pytorch/base pip3 install -r requirements.txt $prefix_sudo python3 setup.py install diff --git a/tests/executables/bert/train_bert_default_amp_dist_1x8_torch.sh b/tests/executables/bert/train_bert_default_amp_dist_1x8_torch.sh index 1eaf33a94..aea938bf8 100644 --- a/tests/executables/bert/train_bert_default_amp_dist_1x8_torch.sh +++ b/tests/executables/bert/train_bert_default_amp_dist_1x8_torch.sh @@ -17,7 +17,7 @@ source ../_utils/global_environment_variables.sh : ${BATCH_SIZE:=10} -cd ../../../nlp/language_model/bert_sample/pytorch/base +cd ../../../models/nlp/language_model/bert_sample/pytorch/base if [ "$?" != "0" ]; then echo "train status: fail." exit 1 diff --git a/tests/executables/bert/train_bert_pretraining_amp_dist_1x8_tf.sh b/tests/executables/bert/train_bert_pretraining_amp_dist_1x8_tf.sh index 6c6b28067..990a798a4 100644 --- a/tests/executables/bert/train_bert_pretraining_amp_dist_1x8_tf.sh +++ b/tests/executables/bert/train_bert_pretraining_amp_dist_1x8_tf.sh @@ -25,7 +25,7 @@ set -euox pipefail current_path=$(cd $(dirname "${BASH_SOURCE[0]}") && pwd) ROOT_DIR=${current_path}"/../../" -SRC_DIR=${ROOT_DIR}../nlp/language_model/bert/tensorflow/base +SRC_DIR=${ROOT_DIR}../models/nlp/language_model/bert/tensorflow/base DATA_DIR=${ROOT_DIR}data/datasets MODEL_DIR=${ROOT_DIR}data/model_zoo diff --git a/tests/executables/bert/train_bert_pretraining_amp_dist_1x8_torch.sh b/tests/executables/bert/train_bert_pretraining_amp_dist_1x8_torch.sh index 6e59480a1..e512b8c9a 100644 --- a/tests/executables/bert/train_bert_pretraining_amp_dist_1x8_torch.sh +++ b/tests/executables/bert/train_bert_pretraining_amp_dist_1x8_torch.sh @@ -15,9 +15,9 @@ set -euox pipefail : ${BATCH_SIZE:=27} -cd ../../../nlp/language_model/bert_sample/pytorch/base/ +cd ../../../models/nlp/language_model/bert_sample/pytorch/base/ if [ "$?" != "0" ]; then - echo "ERROR: ../../../nlp/language_model/bert_sample/pytorch/base/ not exist." + echo "ERROR: ../../../models/nlp/language_model/bert_sample/pytorch/base/ not exist." exit 1 fi diff --git a/tests/executables/conformer/init_torch.sh b/tests/executables/conformer/init_torch.sh index 9635b0bd6..e172c1998 100644 --- a/tests/executables/conformer/init_torch.sh +++ b/tests/executables/conformer/init_torch.sh @@ -1,6 +1,6 @@ #!/bin/bash ROOT_DIR="$(cd "$(dirname "$0")/../.."; pwd)" -SRC_DIR=$ROOT_DIR/../audio/speech_recognition/conformer/pytorch +SRC_DIR=$ROOT_DIR/../models/audio/speech_recognition/conformer/pytorch DATA_DIR=$ROOT_DIR/data # determine whether the user is root mode to execute this script diff --git a/tests/executables/conformer/train_conformer_librispeech_dist_1x8_torch.sh b/tests/executables/conformer/train_conformer_librispeech_dist_1x8_torch.sh index 9b18a1345..95053a9bb 100644 --- a/tests/executables/conformer/train_conformer_librispeech_dist_1x8_torch.sh +++ b/tests/executables/conformer/train_conformer_librispeech_dist_1x8_torch.sh @@ -3,7 +3,7 @@ source ../_utils/global_environment_variables.sh : ${BATCH_SIZE:=8} ROOT_DIR="$(cd "$(dirname "$0")/../.."; pwd)" -SRC_DIR=$ROOT_DIR/../audio/speech_recognition/conformer/pytorch +SRC_DIR=$ROOT_DIR/../models/audio/speech_recognition/conformer/pytorch DATA_DIR=$ROOT_DIR/data export DRT_MEMCPYUSEKERNEL=20000000000 diff --git a/tests/executables/dali/train_resnet50_dali_torch.sh b/tests/executables/dali/train_resnet50_dali_torch.sh index 8f0a5eb8f..3aeba78f0 100644 --- a/tests/executables/dali/train_resnet50_dali_torch.sh +++ b/tests/executables/dali/train_resnet50_dali_torch.sh @@ -10,7 +10,7 @@ fi ixdltest-check --nonstrict_mode_args="--epoch ${NONSTRICT_EPOCH}" -b 8 --run_script \ -python3 ${PROJECT_DIR}../cv/classification/resnet50/pytorch/train.py \ +python3 ${PROJECT_DIR}../models/cv/classification/resnet50/pytorch/train.py \ --data-path ${PROJECT_DIR}/data/datasets/imagenette \ --batch-size ${BATCH_SIZE} \ --output-dir ${OUTPUT_DIR} \ diff --git a/tests/executables/efficientnet/train_efficientb4_imagenet_amp_dist_torch.sh b/tests/executables/efficientnet/train_efficientb4_imagenet_amp_dist_torch.sh index 382e3ec5f..5f6d711f3 100644 --- a/tests/executables/efficientnet/train_efficientb4_imagenet_amp_dist_torch.sh +++ b/tests/executables/efficientnet/train_efficientb4_imagenet_amp_dist_torch.sh @@ -17,7 +17,7 @@ check_status() fi } -cd ../../../cv/classification/efficientnet_b4/pytorch +cd ../../../models/cv/classification/efficientnet_b4/pytorch export PYTHONPATH=./:$PYTHONPATH diff --git a/tests/executables/fairmot/init_torch.sh b/tests/executables/fairmot/init_torch.sh index e01fe08f0..36013ebf7 100644 --- a/tests/executables/fairmot/init_torch.sh +++ b/tests/executables/fairmot/init_torch.sh @@ -7,7 +7,7 @@ unzip -q MOT17.zip mkdir MOT17/images && mkdir MOT17/labels_with_ids mv ./MOT17/train ./MOT17/images/ && mv ./MOT17/test ./MOT17/images/ -cd ${ROOT_DIR}/../cv/multi_object_tracking/fairmot/pytorch/ +cd ${ROOT_DIR}/../models/cv/multi_object_tracking/fairmot/pytorch/ pip3 install Cython pip3 install -r requirements.txt diff --git a/tests/executables/fairmot/train_fairmot_hrnet32_dist_torch.sh b/tests/executables/fairmot/train_fairmot_hrnet32_dist_torch.sh index 2fbf7ccb4..fbf78f2ef 100644 --- a/tests/executables/fairmot/train_fairmot_hrnet32_dist_torch.sh +++ b/tests/executables/fairmot/train_fairmot_hrnet32_dist_torch.sh @@ -14,7 +14,7 @@ fi ROOT_DIR=${CURRENT_DIR}/../.. DADASAT_PATH=${ROOT_DIR}/data/datasets/MOT17 -cd ${ROOT_DIR}/../cv/multi_object_tracking/fairmot/pytorch/ +cd ${ROOT_DIR}/../models/cv/multi_object_tracking/fairmot/pytorch/ bash train_hrnet32_mot17.sh --batch_size $((IX_NUM_CUDA_VISIBLE_DEVICES*BATCH_SIZE)) \ --lr 0.001 \ diff --git a/tests/executables/maskrcnn/init_torch.sh b/tests/executables/maskrcnn/init_torch.sh index 0e6c0becd..6fdd23524 100644 --- a/tests/executables/maskrcnn/init_torch.sh +++ b/tests/executables/maskrcnn/init_torch.sh @@ -8,7 +8,7 @@ if [[ ! -d "${PROJECT_DATA}/VOC2012_sample" ]]; then tar zxf ${PROJECT_DATA}/VOC2012_sample.tgz -C ${PROJECT_DATA} fi -cd ${PROJ_DIR}/cv/detection/maskrcnn/pytorch/ +cd ${PROJ_DIR}/models/cv/detection/maskrcnn/pytorch/ OSNAME=$(cat /proc/version) # install the requirement if [[ "${OSNAME}" == *"aarch64"* ]] diff --git a/tests/executables/maskrcnn/train_maskrcnn_resnet50_amp_torch.sh b/tests/executables/maskrcnn/train_maskrcnn_resnet50_amp_torch.sh index 0ae3a14e5..141ffd47e 100644 --- a/tests/executables/maskrcnn/train_maskrcnn_resnet50_amp_torch.sh +++ b/tests/executables/maskrcnn/train_maskrcnn_resnet50_amp_torch.sh @@ -12,7 +12,7 @@ fi ixdltest-check --nonstrict_mode_args="--epoch ${NONSTRICT_EPOCH}" -b 10 --run_script \ -python3 ${PROJECT_DIR}../cv/detection/maskrcnn/pytorch/train.py \ +python3 ${PROJECT_DIR}../models/cv/detection/maskrcnn/pytorch/train.py \ --model maskrcnn_resnet50_fpn \ --data-path ${PROJECT_DIR}/data/datasets/VOC2012_sample \ --amp \ diff --git a/tests/executables/mobilenetv3/init_torch.sh b/tests/executables/mobilenetv3/init_torch.sh index e10f8246a..f6433d3bb 100644 --- a/tests/executables/mobilenetv3/init_torch.sh +++ b/tests/executables/mobilenetv3/init_torch.sh @@ -12,4 +12,4 @@ echo "prefix_sudo= $prefix_sudo" command -v yum >/dev/null && $prefix_sudo yum install -y numactl || $prefix_sudo apt install -y numactl -pip3 install -r ../../../cv/classification/mobilenetv3/pytorch/requirements.txt \ No newline at end of file +pip3 install -r ../../../models/cv/classification/mobilenetv3/pytorch/requirements.txt \ No newline at end of file diff --git a/tests/executables/mobilenetv3/train_mobilenetv3_large_amp_torch.sh b/tests/executables/mobilenetv3/train_mobilenetv3_large_amp_torch.sh index f50b8e88e..5bf7a37ad 100644 --- a/tests/executables/mobilenetv3/train_mobilenetv3_large_amp_torch.sh +++ b/tests/executables/mobilenetv3/train_mobilenetv3_large_amp_torch.sh @@ -16,7 +16,7 @@ check_status() fi } -cd $CURRENT_DIR/../../../cv/classification/mobilenetv3/pytorch/ +cd $CURRENT_DIR/../../../models/cv/classification/mobilenetv3/pytorch/ ixdltest-check --nonstrict_mode_args="--epoch ${NONSTRICT_EPOCH}" -b 10 --run_script \ python3 train.py --model mobilenet_v3_large --data-path "${DATA_DIR}" \ --epochs 600 --batch-size ${BATCH_SIZE} --opt sgd --lr 0.1 \ diff --git a/tests/executables/resnet/init_paddle.sh b/tests/executables/resnet/init_paddle.sh index 70773191f..152dae2a9 100644 --- a/tests/executables/resnet/init_paddle.sh +++ b/tests/executables/resnet/init_paddle.sh @@ -13,7 +13,7 @@ if [ ! -d "${DATASET_DIR}/flowers102" ]; then tar zxf ${DATASET_DIR}/flowers102.tgz -C ${DATASET_DIR} fi -RESNET_PADDLE_DIR=${PRJ_DIR}/../cv/classification/resnet50/paddlepaddle +RESNET_PADDLE_DIR=${PRJ_DIR}/../models/cv/classification/resnet50/paddlepaddle cd ${RESNET_PADDLE_DIR} pip3 install -r requirements.txt diff --git a/tests/executables/resnet/train_resnet50_amp_torch.sh b/tests/executables/resnet/train_resnet50_amp_torch.sh index 46f15ff50..387014b60 100644 --- a/tests/executables/resnet/train_resnet50_amp_torch.sh +++ b/tests/executables/resnet/train_resnet50_amp_torch.sh @@ -9,7 +9,7 @@ if [[ -d ${OUTPUT_DIR} ]]; then fi ixdltest-check --nonstrict_mode_args="--epoch ${NONSTRICT_EPOCH}" -b 8 --run_script \ -python3 ${PROJECT_DIR}../cv/classification/resnet50/pytorch/train.py \ +python3 ${PROJECT_DIR}../models/cv/classification/resnet50/pytorch/train.py \ --data-path ${PROJECT_DIR}/data/datasets/imagenette \ --batch-size ${BATCH_SIZE} \ --lr 0.01 \ diff --git a/tests/executables/resnet/train_resnet50_dist_paddle.sh b/tests/executables/resnet/train_resnet50_dist_paddle.sh index 777cf7c17..14ce65b00 100644 --- a/tests/executables/resnet/train_resnet50_dist_paddle.sh +++ b/tests/executables/resnet/train_resnet50_dist_paddle.sh @@ -8,7 +8,7 @@ if [[ -d ${OUTPUT_DIR} ]]; then mkdir -p ${OUTPUT_DIR} fi -RESNET_PADDLE_DIR=${PROJECT_DIR}../cv/classification/resnet50/paddlepaddle/ +RESNET_PADDLE_DIR=${PROJECT_DIR}../models/cv/classification/resnet50/paddlepaddle/ cd ${RESNET_PADDLE_DIR} ixdltest-check --nonstrict_mode_args="--epoch ${NONSTRICT_EPOCH}" -b 8 --run_script \ diff --git a/tests/executables/resnet/train_resnet50_dist_tf.sh b/tests/executables/resnet/train_resnet50_dist_tf.sh index a71d21bb6..f3a0c2ebb 100644 --- a/tests/executables/resnet/train_resnet50_dist_tf.sh +++ b/tests/executables/resnet/train_resnet50_dist_tf.sh @@ -16,7 +16,7 @@ check_status() fi } -cd ${ROOT_DIR}/../cv/classification/resnet50/tensorflow/ +cd ${ROOT_DIR}/../models/cv/classification/resnet50/tensorflow/ ixdltest-check --nonstrict_mode_args="--epoch ${NONSTRICT_EPOCH}" -b 0.01 --run_script \ bash run_train_resnet50_distributed_imagenette.sh "$@"; check_status diff --git a/tests/executables/retinanet/train_retinanet_amp_torch.sh b/tests/executables/retinanet/train_retinanet_amp_torch.sh index d64bfe4fb..937faf3b4 100644 --- a/tests/executables/retinanet/train_retinanet_amp_torch.sh +++ b/tests/executables/retinanet/train_retinanet_amp_torch.sh @@ -16,7 +16,7 @@ check_status() } ixdltest-check --nonstrict_mode_args="--epoch ${NONSTRICT_EPOCH}" -b 0 --run_script \ -python3 ../../../cv/detection/retinanet/pytorch/train.py \ +python3 ../../../models/cv/detection/retinanet/pytorch/train.py \ --model retinanet_resnet50_fpn \ --lr 0.01 \ --data-path ${DATA_DIR} \ diff --git a/tests/executables/ssd/init_tf.sh b/tests/executables/ssd/init_tf.sh index 51111a05b..33847fdd9 100644 --- a/tests/executables/ssd/init_tf.sh +++ b/tests/executables/ssd/init_tf.sh @@ -4,7 +4,7 @@ PROJECT_ROOT="${CUR_DIR}/../.." DATASET_DIR="${PROJECT_ROOT}/data/datasets" MODEL_CPT_DIR="${PROJECT_ROOT}/data/model_zoo/ssd_tf" VOC_RECORD_DIR="${DATASET_DIR}/tf_ssd_voc_record" -SSD_ROOT="${PROJECT_ROOT}/../cv/detection/ssd/tensorflow" +SSD_ROOT="${PROJECT_ROOT}/../models/cv/detection/ssd/tensorflow" # determine whether the user is root mode to execute this script prefix_sudo="" diff --git a/tests/executables/ssd/init_torch.sh b/tests/executables/ssd/init_torch.sh index c4fd1c84e..c8784bb04 100644 --- a/tests/executables/ssd/init_torch.sh +++ b/tests/executables/ssd/init_torch.sh @@ -51,7 +51,7 @@ if [[ "$(uname -m)" == "aarch64" ]]; then source /opt/rh/gcc-toolset-11/enable fi -cd ../../../cv/detection/ssd/pytorch/ && bash ./clean_ssd.sh && bash ./build_ssd.sh && bash ./install_ssd.sh "$@"; check_status +cd ../../../models/cv/detection/ssd/pytorch/ && bash ./clean_ssd.sh && bash ./build_ssd.sh && bash ./install_ssd.sh "$@"; check_status DATA_PATH_BBOX=../../../.. python3 prepare-json.py --keep-keys ${DATA_PATH}/annotations/instances_val2017.json ${DATA_PATH_BBOX}/bbox_only_instances_val2017.json "$@"; check_status diff --git a/tests/executables/ssd/train_ssd_amp_tf.sh b/tests/executables/ssd/train_ssd_amp_tf.sh index d3b6050cc..204044001 100644 --- a/tests/executables/ssd/train_ssd_amp_tf.sh +++ b/tests/executables/ssd/train_ssd_amp_tf.sh @@ -10,7 +10,7 @@ ROOT_DIR="${CURRENT_DIR}/../.." DATASET_PATH="${ROOT_DIR}/data/datasets/imagenette" MODEL_ZOO="${ROOT_DIR}/data/model_zoo" WORKSPACE="${ROOT_DIR}/output/${MODEL_NAME}/$0/${CURTIME}" -SRC_DIR="${ROOT_DIR}/../cv/detection/ssd/tensorflow" +SRC_DIR="${ROOT_DIR}/../models/cv/detection/ssd/tensorflow" EXIT_STATUS=0 check_status() diff --git a/tests/executables/ssd/train_ssd_amp_torch.sh b/tests/executables/ssd/train_ssd_amp_torch.sh index e6c3bea77..e9262956b 100644 --- a/tests/executables/ssd/train_ssd_amp_torch.sh +++ b/tests/executables/ssd/train_ssd_amp_torch.sh @@ -11,7 +11,7 @@ check_status() fi } -cd ../../../cv/detection/ssd/pytorch/ +cd ../../../models/cv/detection/ssd/pytorch/ echo "python3 train.py --no-dali --dali-cache 0 --data=${COCO_PATH} \ --batch-size=${BATCH_SIZE} --warmup-factor=0 --warmup=650 --lr=2.92e-3 --threshold=0.08 --epochs 5 --eval-batch-size=160 \ diff --git a/tests/executables/stable-diffusion/init_torch.sh b/tests/executables/stable-diffusion/init_torch.sh index ea0bb7a09..99cb54a06 100644 --- a/tests/executables/stable-diffusion/init_torch.sh +++ b/tests/executables/stable-diffusion/init_torch.sh @@ -1,6 +1,6 @@ #!/bin/bash ROOT_DIR="$(cd "$(dirname "$0")/../.."; pwd)" -SRC_DIR=$ROOT_DIR/../multimodal/diffusion_model/stable-diffusion-2.1-pokemon/pytorch +SRC_DIR=$ROOT_DIR/../models/multimodal/diffusion_model/stable-diffusion-2.1-pokemon/pytorch DATA_DIR=$ROOT_DIR/data # install packages diff --git a/tests/executables/stable-diffusion/train_sd2.1_pokemon_dist_1x8_torch.sh b/tests/executables/stable-diffusion/train_sd2.1_pokemon_dist_1x8_torch.sh index d19bab854..ff9ed42ae 100644 --- a/tests/executables/stable-diffusion/train_sd2.1_pokemon_dist_1x8_torch.sh +++ b/tests/executables/stable-diffusion/train_sd2.1_pokemon_dist_1x8_torch.sh @@ -6,7 +6,7 @@ export ENABLE_FLASH_ATTENTION_WITH_IXATTNBKD=0 : ${BATCH_SIZE:=8} ROOT_DIR="$(cd "$(dirname "$0")/../.."; pwd)" -SRC_DIR=$ROOT_DIR/../multimodal/diffusion_model/stable-diffusion-2.1-pokemon/pytorch/examples/text_to_image +SRC_DIR=$ROOT_DIR/../models/multimodal/diffusion_model/stable-diffusion-2.1-pokemon/pytorch/examples/text_to_image DATASET_NAME=$ROOT_DIR/data/datasets/pokemon-blip-captions MODEL_NAME=$ROOT_DIR/data/model_zoo/stabilityai/stable-diffusion-2-1 export DRT_MEMCPYUSEKERNEL=20000000000 diff --git a/tests/executables/unet3d/train_unet3d_kits19_stage3_dist_1x8_torch.sh b/tests/executables/unet3d/train_unet3d_kits19_stage3_dist_1x8_torch.sh index 8c581e6e9..6de759b9a 100644 --- a/tests/executables/unet3d/train_unet3d_kits19_stage3_dist_1x8_torch.sh +++ b/tests/executables/unet3d/train_unet3d_kits19_stage3_dist_1x8_torch.sh @@ -18,7 +18,7 @@ GRADIENT_ACCUMULATION_STEPS=1 SAVE_CKPT="./ckpt_stage3" LOG_NAME='train_log_stage3.json' -cd ../../../cv/semantic_segmentation/unet3d/pytorch +cd ../../../models/cv/semantic_segmentation/unet3d/pytorch if [ ! -d ${SAVE_CKPT} ]; then mkdir ${SAVE_CKPT}; fi diff --git a/tests/executables/yolov5/init_torch.sh b/tests/executables/yolov5/init_torch.sh index 24efe516a..47e5cae8b 100644 --- a/tests/executables/yolov5/init_torch.sh +++ b/tests/executables/yolov5/init_torch.sh @@ -14,7 +14,7 @@ pip3 install tqdm==4.62.1 git clone https://gitee.com/deep-spark/deepsparkhub-gpl.git ../../deepsparkhub-gpl -cd ${PROJ_DIR}/deepsparkhub-gpl/cv/detection/yolov5-sample/pytorch +cd ${PROJ_DIR}/deepsparkhub-gpl/models/cv/detection/yolov5-sample/pytorch # Remove exist datas if [[ -d "./datasets/coco128" ]]; then diff --git a/tests/executables/yolov5/train_yolov5s_coco_amp_torch.sh b/tests/executables/yolov5/train_yolov5s_coco_amp_torch.sh index 471532b6a..099672579 100644 --- a/tests/executables/yolov5/train_yolov5s_coco_amp_torch.sh +++ b/tests/executables/yolov5/train_yolov5s_coco_amp_torch.sh @@ -14,7 +14,7 @@ check_status() fi } -cd "${ROOT_DIR}/deepsparkhub-gpl/cv/detection/yolov5-sample/pytorch" +cd "${ROOT_DIR}/deepsparkhub-gpl/models/cv/detection/yolov5-sample/pytorch" ixdltest-check --nonstrict_mode_args="--epoch 1" -b 0.2 --run_script \ python3 train.py --img-size 640 --batch-size ${BATCH_SIZE} \ --cfg ./models/yolov5s.yaml --weights ./weights/yolov5s.pt --data ./data/coco.yaml --amp ${nonstrict_mode_args} "$@"; check_status diff --git a/tests/run_test.py b/tests/run_test.py index 081ee8e26..3a7a9e05e 100644 --- a/tests/run_test.py +++ b/tests/run_test.py @@ -112,7 +112,7 @@ def run_detec_testcase(model): # 选择使用atss作为个例 prepare_script = f""" apt install -y libgl1-mesa-glx - cd ../cv/detection/atss_mmdet/pytorch/ + cd ../models/cv/detection/atss_mmdet/pytorch/ cp -r /mnt/deepspark/data/3rd_party/mmdetection-v3.3.0 ./mmdetection > /dev/null 2>&1 mkdir -p /root/.cache/torch/hub/checkpoints/ cp /mnt/deepspark/data/checkpoints/resnet50-0676ba61.pth /root/.cache/torch/hub/checkpoints/resnet50-0676ba61.pth @@ -127,7 +127,7 @@ def run_detec_testcase(model): # 选择使用CenterNet作为个例 prepare_script = f""" apt install -y libgl1-mesa-glx - cd ../cv/detection/centernet/paddlepaddle/ + cd ../models/cv/detection/centernet/paddlepaddle/ cp -r /mnt/deepspark/data/3rd_party/PaddleDetection-release-2.6 ./PaddleDetection cd PaddleDetection pip install -r requirements.txt @@ -139,7 +139,7 @@ def run_detec_testcase(model): elif is_yolov: # 选择使用yolov5作为个例 prepare_script = f""" - cd ../cv/detection/yolov5/pytorch/ + cd ../models/cv/detection/yolov5/pytorch/ bash ci/prepare.sh """ else: @@ -212,7 +212,7 @@ def run_clf_testcase(model): # 选择使用mocov2作为个例 prepare_script = f""" apt install -y libgl1-mesa-glx - cd ../cv/classification/mocov2/pytorch/ + cd ../models/cv/classification/mocov2/pytorch/ cp -r /mnt/deepspark/data/3rd_party/mmpretrain-v1.2.0 ./mmpretrain > /dev/null 2>&1 cd mmpretrain python3 setup.py install @@ -226,7 +226,7 @@ def run_clf_testcase(model): # 选择使用googlenet作为个例 prepare_script = f""" apt install -y libgl1-mesa-glx - cd ../cv/classification/googlenet/paddlepaddle/ + cd ../models/cv/classification/googlenet/paddlepaddle/ cp -r /mnt/deepspark/data/3rd_party/PaddleClas-release-2.6 ./PaddleClas > /dev/null 2>&1 cd PaddleClas pip3 install -r requirements.txt @@ -239,7 +239,7 @@ def run_clf_testcase(model): # 选择使用alexnet作为个例 prepare_script = f""" apt install -y libgl1-mesa-glx - cd ../cv/classification/alexnet/tensorflow/ + cd ../models/cv/classification/alexnet/tensorflow/ ln -s /mnt/deepspark/data/datasets/imagenet_tfrecord/ILSVRC2012 dataset/imagenet_tfrecord timeout 1800 bash run_train_alexnet_imagenet.sh """ -- Gitee