Merge branch 'PaddlePaddle:dygraph' into dygraph

7c7972f3 · topduke · GitHub · f5f8af4d · 79640f5d · 7c7972f3
Unverified Commit 7c7972f3 authored Apr 28, 2022 by topduke Committed by GitHub Apr 28, 2022
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

tools/infer_cls.py tools/infer_cls.py +2 -0

No files found.
--- a/tools/infer_cls.py
+++ b/tools/infer_cls.py
@@ -57,6 +57,8 @@ def main():
            continue
        elif op_name == 'KeepKeys':
            op[op_name]['keep_keys'] = ['image']
+        elif op_name == "SSLRotateResize":
+            op[op_name]["mode"] = "test"
        transforms.append(op)
    global_config['infer_mode'] = True
    ops = create_operators(transforms, global_config)