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

fix deeptext 310 datatype bug

This commit is contained in:
yuzhenhua 2021-04-14 10:14:03 +08:00
parent f3ccaed062
commit d2654b6d23
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)