Skip to content

GitLab

"src/git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "f40aee9d2e69fab1262d4cb55ea4dc4863ddccdb"
Commit 8e697e34 authored by WenmuZhou's avatar WenmuZhou
Browse files

add note

parent b30add8a
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
tools/export_model.py
View file @ 8e697e34
...@@ -54,10 +54,13 @@ def main(): ...@@ -54,10 +54,13 @@ def main():
infer_shape = [3, -1, -1] infer_shape = [3, -1, -1]
if config['Architecture']['model_type'] == "rec": 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'][ if 'Transform' in config['Architecture'] and config['Architecture'][
'Transform'] is not None and config['Architecture'][ 'Transform'] is not None and config['Architecture'][
'Transform']['name'] == 'TPS': '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 infer_shape[-1] = 100
model = to_static( model = to_static(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment