diff --git a/TensorFlow/built-in/cv/detection/MaskRcnn_ID0011_for_TensorFlow/test/train_full_16p.sh b/TensorFlow/built-in/cv/detection/MaskRcnn_ID0011_for_TensorFlow/test/train_full_16p.sh index 049b4fae8879e452ece702cd7185d1c78de9f004..5f17c596d7e49a9c02013f82d0bc3afb54785004 100644 --- a/TensorFlow/built-in/cv/detection/MaskRcnn_ID0011_for_TensorFlow/test/train_full_16p.sh +++ b/TensorFlow/built-in/cv/detection/MaskRcnn_ID0011_for_TensorFlow/test/train_full_16p.sh @@ -169,7 +169,7 @@ do --validation_file_pattern=${data_path}/val* \ --val_json_file=${data_path}/instances_val2017.json \ --eval_batch_size=2 \ - --model_dir=result_npu\ + --model_dir=${cur_path}/output/${ASCEND_DEVICE_ID}/ckpt \ --over_dump=${over_dump} \ --over_dump_path=${over_dump_path} \ > ${cur_path}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 & diff --git a/TensorFlow/built-in/cv/detection/MaskRcnn_ID0011_for_TensorFlow/test/train_full_8p.sh b/TensorFlow/built-in/cv/detection/MaskRcnn_ID0011_for_TensorFlow/test/train_full_8p.sh index 3b3d47825dc6208386626d783fad1044351c619a..b144fbee96c0992fb7384075a5da93c7b8b15cfe 100644 --- a/TensorFlow/built-in/cv/detection/MaskRcnn_ID0011_for_TensorFlow/test/train_full_8p.sh +++ b/TensorFlow/built-in/cv/detection/MaskRcnn_ID0011_for_TensorFlow/test/train_full_8p.sh @@ -153,7 +153,7 @@ do --validation_file_pattern=${data_path}/val* \ --val_json_file=${data_path}/instances_val2017.json \ --eval_batch_size=2 \ - --model_dir=result_npu\ + --model_dir=${cur_path}/output/${ASCEND_DEVICE_ID}/ckpt \ --over_dump=${over_dump} \ --over_dump_path=${over_dump_path} \ > ${cur_path}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 & diff --git a/TensorFlow/built-in/cv/detection/MaskRcnn_ID0011_for_TensorFlow/test/train_performance_8p.sh b/TensorFlow/built-in/cv/detection/MaskRcnn_ID0011_for_TensorFlow/test/train_performance_8p.sh index abad3802497b11aba5e950c5587d28a21253a837..3b54089aa4d8e54e9a671605fcc1c97f0d0500c0 100644 --- a/TensorFlow/built-in/cv/detection/MaskRcnn_ID0011_for_TensorFlow/test/train_performance_8p.sh +++ b/TensorFlow/built-in/cv/detection/MaskRcnn_ID0011_for_TensorFlow/test/train_performance_8p.sh @@ -139,7 +139,7 @@ do mkdir -p ${cur_path}/output/$ASCEND_DEVICE_ID/ckpt fi - + #执行训练脚本,以下传参不需要修改,其他需要模型审视修改 #--data_dir, --model_dir, --precision_mode, --over_dump, --over_dump_path,--data_dump_flag,--data_dump_step,--data_dump_path,--profiling,--profiling_dump_path @@ -159,7 +159,7 @@ do --validation_file_pattern=${data_path}/val* \ --val_json_file=${data_path}/instances_val2017.json \ --eval_batch_size=2 \ - --model_dir=result_npu\ + --model_dir=${cur_path}/output/${ASCEND_DEVICE_ID}/ckpt \ --over_dump=${over_dump} \ --over_dump_path=${over_dump_path} \ > ${cur_path}/output/${ASCEND_DEVICE_ID}/train_${ASCEND_DEVICE_ID}.log 2>&1 &