Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "0888a4ae2bff1098ba61375f43c30d5234a52ed9"
Unverified Commit 7c7972f3 authored by topduke's avatar topduke Committed by GitHub
Browse files

Merge branch 'PaddlePaddle:dygraph' into dygraph

parents f5f8af4d 79640f5d
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tools/infer_cls.py
View file @ 7c7972f3
...@@ -57,6 +57,8 @@ def main(): ...@@ -57,6 +57,8 @@ def main():
continue continue
elif op_name == 'KeepKeys': elif op_name == 'KeepKeys':
op[op_name]['keep_keys'] = ['image'] op[op_name]['keep_keys'] = ['image']
elif op_name == "SSLRotateResize":
op[op_name]["mode"] = "test"
transforms.append(op) transforms.append(op)
global_config['infer_mode'] = True global_config['infer_mode'] = True
ops = create_operators(transforms, global_config) ops = create_operators(transforms, global_config)
......
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