Skip to content

GitLab

"tests/git@developer.sourcefind.cn:OpenDAS/fastfold.git" did not exist on "64f5f84706c7d6076680a3230eba3ad5a154d455"
Unverified Commit eed3d6aa authored by andyjpaddle's avatar andyjpaddle Committed by GitHub
Browse files

Merge pull request #7232 from tink2123/revert_imgh_2.5 

rever rec input h
parents 93b49541 0b97b0c4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/export_model.py
View file @ eed3d6aa
...@@ -83,7 +83,7 @@ def export_single_model(model, ...@@ -83,7 +83,7 @@ def export_single_model(model,
else: else:
infer_shape = [3, -1, -1] infer_shape = [3, -1, -1]
if arch_config["model_type"] == "rec": if arch_config["model_type"] == "rec":
infer_shape = [3, 48, -1] # for rec model, H must be 32 infer_shape = [3, 32, -1] # for rec model, H must be 32
if "Transform" in arch_config and arch_config[ if "Transform" in arch_config and arch_config[
"Transform"] is not None and arch_config["Transform"][ "Transform"] is not None and arch_config["Transform"][
"name"] == "TPS": "name"] == "TPS":
......
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