Update benchmark_utils.py

dmlc · Aug 10, 2020 · 9e0f631 · 9e0f631
1 parent 37f224f
commit 9e0f631
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/benchmarks/benchmark_utils.py b/scripts/benchmarks/benchmark_utils.py
@@ -1003,7 +1003,7 @@ def save_to_csv(self, result_dict, filename):
                             "model": model_name,
                             "batch_size": bs,
                             "sequence_length": ss,
-                            'latency': str(int(latency * 1000)) if not np.isnan(latency) else str(np.nan),
+                            'latency': str(round(latency * 1000)) if not np.isnan(latency) else str(np.nan),
                             'memory': str(memory),
                         }
                     )