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

wgan export bug fix master

This commit is contained in:
zhangxiaoxiao 2021-07-06 18:17:30 +08:00
parent 45d3f32f9d
commit 0e98f233d9
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
model_zoo/research/cv/wgan/export.py
View File

@ -23,7 +23,8 @@ import mindspore.common.dtype as mstype
from mindspore import Tensor, load_checkpoint, load_param_into_net, export, context from mindspore import Tensor, load_checkpoint, load_param_into_net, export, context
from src.args import get_args from src.args import get_args
from src.dcgan_model import DcganG
from src.dcgannobn_model import DcgannobnG
if __name__ == '__main__': if __name__ == '__main__':
args_opt = get_args('export') args_opt = get_args('export')

4
model_zoo/research/cv/wgan/scripts/run_infer_310.sh
View File

@ -102,6 +102,10 @@ function infer()
function cal_acc() function cal_acc()
{ {
if [ -d infer_output ]; then
rm -rf ./infer_output
fi
mkdir infer_output
python3.7 ../postprocess.py --config=$config_path --output_dir=./infer_output --nimages=$nimages --post_result_path=./result_Files --device_id=$device_id &> acc.log python3.7 ../postprocess.py --config=$config_path --output_dir=./infer_output --nimages=$nimages --post_result_path=./result_Files --device_id=$device_id &> acc.log
} }