diff --git a/research/cv/ras/export.py b/research/cv/ras/export.py index 9c2ba36c9ae5442f3787cd60da6bd0bf7044770f..8ead39b3f192326dd1ca042143036411c31f528c 100644 --- a/research/cv/ras/export.py +++ b/research/cv/ras/export.py @@ -19,7 +19,7 @@ import argparse import numpy as np import mindspore as ms from mindspore import load_checkpoint, load_param_into_net, export -from .src.model import BoneModel +from src.model import BoneModel def run_export(device_target, device_id, pretrained_model, model_ckpt, batchsize): diff --git a/research/cv/vit_base/scripts/run_distribute_train_ascend.sh b/research/cv/vit_base/scripts/run_distribute_train_ascend.sh index d9be2187f704e70eed16c4a7cd7ab34ac5c217f3..28569a25e86caf295543b1f926ec28f328283c62 100644 --- a/research/cv/vit_base/scripts/run_distribute_train_ascend.sh +++ b/research/cv/vit_base/scripts/run_distribute_train_ascend.sh @@ -26,10 +26,9 @@ RANK_TABLE_FILE=$(realpath $1) export RANK_TABLE_FILE echo "RANK_TABLE_FILE=${RANK_TABLE_FILE}" -device_start=$3 for((i=0; i<${DEVICE_NUM}; i++)) do - export DEVICE_ID=$((device_start + i)) + export DEVICE_ID=$i export RANK_ID=$i rm -rf ./train_parallel$i mkdir ./train_parallel$i