p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix data_path conflict

This commit is contained in:
l00486551 2021-06-24 15:19:23 +08:00
parent 70152adcb3
commit 8d4671b04e
3 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
model_zoo/official/cv/tinydarknet/cifar10_config.yaml
View File

@ -26,8 +26,8 @@ momentum: 0.9
weight_decay: 0.0001
image_height: 227
image_width: 227
train_data_dir: './dataset/imagenet_original/train/'
val_data_dir: './dataset/imagenet_original/val/'
train_data_dir: './data/cifar10_train/'
val_data_dir: './data/cifar10_val/'
keep_checkpoint_max: 1
checkpoint_path: './scripts/train_parallel4/ckpt_4/train_tinydarknet_imagenet-300_1251.ckpt'
onnx_filename: 'tinydarknet.onnx'

2
model_zoo/official/cv/tinydarknet/scripts/run_train_cpu.sh
View File

@ -53,5 +53,5 @@ cp ./*.yaml ./train_cpu
echo "start training for device CPU"
cd ./train_cpu || exit
env > env.log
python train.py --device_target=CPU --data_path=$PATH1 --dataset_name=$2 --config_path=$CONFIG_FILE --lr_init=0.01> ./train.log 2>&1 &
python train.py --device_target=CPU --train_data_dir=$PATH1 --dataset_name=$2 --config_path=$CONFIG_FILE> ./train.log 2>&1 &
cd ..

4
model_zoo/official/cv/tinydarknet/train.py
View File

@ -121,9 +121,9 @@ def modelarts_pre_process():
@moxing_wrapper(pre_process=modelarts_pre_process)
def run_train():
if config.dataset_name == "imagenet":
dataset = create_dataset_imagenet(config.data_path, 1)
dataset = create_dataset_imagenet(config.train_data_dir, 1)
elif config.dataset_name == "cifar10":
dataset = create_dataset_cifar(dataset_path=config.data_path,
dataset = create_dataset_cifar(dataset_path=config.train_data_dir,
do_train=True,
repeat_num=1,
batch_size=config.batch_size,