Fix example & doc table

Change-Id: I4b0707f7c8c002056211d6873c5345a01b7975be
Jeffwhen · Aug 2, 2022 · 5e94d1c · 5e94d1c
1 parent 9410755
commit 5e94d1c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -42,12 +42,14 @@ python3 -m tpu_perf.precision_benchmark # To run precision benchmark
 #### Common variables
 
 | Field                 | Type      | Usage                                                                         |
+|-----------------------|-----------|-------------------------------------------------------------------------------|
 | name                  | Required  | Specify network name, should be unique                                        |
 | gops                  | Optional  | Specify network FLOPs                                                         |
 
 #### NNTC
 
 | Field                     | Type      | Usage                                                                                     |
+|---------------------------|-----------|-------------------------------------------------------------------------------------------|
 | precision                 | Optional  | Boolean type. Indicates whether to build precision model when `tpu_perf.build` is called. |
 | fp32\_compile\_options    | Optional  | Will use this command to build FP32 bmodel.                                               |
 | cali                      | Optional  | Calibration command. `tpu_perf.build` will do calibrating if this variable exists.        |

diff --git a/example/pnet/config.yaml b/example/pnet/config.yaml
@@ -16,7 +16,7 @@ fp32_compile_options:
   --opt 1
   --cmp 1
 
-cali:
+time_only_cali:
   python3 -m ufw.cali.cali_model
     --model $(model)
     --weights $(weight)

diff --git a/python/tpu_perf/run.py b/python/tpu_perf/run.py
@@ -86,10 +86,10 @@ def run(tree, path, config, stat_f):
         if os.path.exists(profile_path):
             info = parse_profile(profile_path)
             rounds = int(1200 / info['runtime'])
-            logging.info(f'Run {rounds} times for {name}.{b}b')
-        max_rounds = 500000
+        max_rounds = 10000
         if rounds > max_rounds:
             rounds = max_rounds
+        logging.info(f'Run {rounds} times for {name}.{b}b')
 
         title = f'run.{b}'
         ref_fn = os.path.join(bmodel_dir, 'output_ref_data.dat')