From d2654b6d232a2b186116151af924bc458f602e77 Mon Sep 17 00:00:00 2001 From: yuzhenhua Date: Wed, 14 Apr 2021 10:14:03 +0800 Subject: [PATCH] fix deeptext 310 datatype bug --- model_zoo/official/cv/deeptext/export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo/official/cv/deeptext/export.py b/model_zoo/official/cv/deeptext/export.py index acbf3d7229..ac099afd7a 100644 --- a/model_zoo/official/cv/deeptext/export.py +++ b/model_zoo/official/cv/deeptext/export.py @@ -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)