Skip to content
Snippets Groups Projects
Commit a8020515 authored by anzhengqi's avatar anzhengqi
Browse files

modify vit_base network

parent 15ccc378
No related branches found
No related tags found
No related merge requests found
...@@ -19,7 +19,7 @@ import argparse ...@@ -19,7 +19,7 @@ import argparse
import numpy as np import numpy as np
import mindspore as ms import mindspore as ms
from mindspore import load_checkpoint, load_param_into_net, export 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): def run_export(device_target, device_id, pretrained_model, model_ckpt, batchsize):
......
...@@ -26,10 +26,9 @@ RANK_TABLE_FILE=$(realpath $1) ...@@ -26,10 +26,9 @@ RANK_TABLE_FILE=$(realpath $1)
export RANK_TABLE_FILE export RANK_TABLE_FILE
echo "RANK_TABLE_FILE=${RANK_TABLE_FILE}" echo "RANK_TABLE_FILE=${RANK_TABLE_FILE}"
device_start=$3
for((i=0; i<${DEVICE_NUM}; i++)) for((i=0; i<${DEVICE_NUM}; i++))
do do
export DEVICE_ID=$((device_start + i)) export DEVICE_ID=$i
export RANK_ID=$i export RANK_ID=$i
rm -rf ./train_parallel$i rm -rf ./train_parallel$i
mkdir ./train_parallel$i mkdir ./train_parallel$i
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment