From a8020515f41e7a1166e4f76676d14d9eda910fcc Mon Sep 17 00:00:00 2001
From: anzhengqi <anzhengqi1@huawei.com>
Date: Sat, 25 Jun 2022 17:59:55 +0800
Subject: [PATCH] modify vit_base network
---
research/cv/ras/export.py | 2 +-
research/cv/vit_base/scripts/run_distribute_train_ascend.sh | 3 +--
2 files changed, 2 insertions(+), 3 deletions(-)
diff --git a/research/cv/ras/export.py b/research/cv/ras/export.py
index 9c2ba36c9..8ead39b3f 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 d9be2187f..28569a25e 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
--
GitLab