From 8e697e349fdb17d7fb7148ed1f79bec08936e207 Mon Sep 17 00:00:00 2001 From: WenmuZhou Date: Thu, 28 Jan 2021 13:21:10 +0800 Subject: [PATCH] add note --- tools/export_model.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tools/export_model.py b/tools/export_model.py index 542dffe3..a9b9e7dd 100755 --- a/tools/export_model.py +++ b/tools/export_model.py @@ -54,10 +54,13 @@ def main(): infer_shape = [3, -1, -1] if config['Architecture']['model_type'] == "rec": - infer_shape = [3, 32, -1] + infer_shape = [3, 32, -1] # for rec model, H must be 32 if 'Transform' in config['Architecture'] and config['Architecture'][ 'Transform'] is not None and config['Architecture'][ 'Transform']['name'] == 'TPS': + logger.info( + 'When there is tps in the network, variable length input is not supported, and the input size needs to be the same as during training' + ) infer_shape[-1] = 100 model = to_static( -- GitLab