Fix fp_batch_sizes not working

Change-Id: Ia6765812f5d72cc8e0b8d674bc67d7d73cd85f1d
Jeffwhen · Nov 16, 2022 · a783ed3 · a783ed3
1 parent 8a098db
commit a783ed3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tpu_perf/build.py b/python/tpu_perf/build.py
@@ -115,7 +115,7 @@ def build_nntc(tree, path, config):
         start = time.monotonic()
         logging.info(f'Building float bmodel {name}...')
         batch_sizes = config.get('fp_batch_sizes', [1]) \
-            if not option_time_only else [1]
+            if option_time_only else [1]
         fp_loops = config.get('fp_loops') or \
             tree.global_config.get('fp_loops') or [dict()]
         for loop in fp_loops: