mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!15133 fix DeepText 310 inference datatype bug 

From: @yuzhenhua666
Reviewed-by: @oacjiewen,@c_34
Signed-off-by: @c_34
This commit is contained in:
mindspore-ci-bot 2021-04-14 14:51:20 +08:00 committed by Gitee
parent 8bf983c58d d2654b6d23
commit 7211f6f1f4
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/deeptext/export.py
View File

@ -48,6 +48,6 @@ if __name__ == '__main__':
load_param_into_net(net, param_dict_new)
img_data = Tensor(np.zeros([config.test_batch_size, 3, config.img_height, config.img_width]), ms.float16)
img_data = Tensor(np.zeros([config.test_batch_size, 3, config.img_height, config.img_width]), ms.float32)
export(net, img_data, file_name=args.file_name, file_format=args.file_format)