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

!7296 fix 310 psenet 

Merge pull request !7296 from anzhengqi/fix-310-psenet
This commit is contained in:
mindspore-ci-bot 2020-10-15 09:18:53 +08:00 committed by Gitee
parent ee53c4079d 0347d4e094
commit f363a41543
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/cv/psenet/export.py
View File

@ -36,4 +36,4 @@ if __name__ == '__main__':
input_arr = Tensor(np.random.uniform(0.0, 1.0, size=[1, 3, config.INFER_LONG_SIZE, config.INFER_LONG_SIZE]), input_arr = Tensor(np.random.uniform(0.0, 1.0, size=[1, 3, config.INFER_LONG_SIZE, config.INFER_LONG_SIZE]),
ms.float32) ms.float32)
export(net, input_arr, file_name=config.geir_filename) export(net, input_arr, file_name=config.air_filename)