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)