Browse Source

!15133 fix DeepText 310 inference datatype bug

From: @yuzhenhua666
Reviewed-by: @oacjiewen,@c_34
Signed-off-by: @c_34
pull/15133/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
7211f6f1f4
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model_zoo/official/cv/deeptext/export.py

+ 1
- 1
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)

Loading…
Cancel
Save