bug fix

2026-09-10 11:50:20 +08:00 · 2021-03-18 13:33:23 +01:00
parent f3e5ddfaaf
commit 6e68637f48
1 changed files with 1 additions and 1 deletions
@@ -176,7 +176,7 @@ def process_args(args, model_type):

    _ = os.path.dirname(os.path.realpath(__file__))

-    if c.mixed_precision:
+    if 'mixed_precision' in c and c.mixed_precision:
        print("   >  Mixed precision mode is ON")

    out_path = args.continue_path