append unit time after the value

tested locally, worked
This commit is contained in:
Albert Angles 2022-12-28 15:42:47 +01:00
parent f25bdd1443
commit 428b8cafdc

View File

@ -55,9 +55,9 @@ def aggregate_stats(stats: dict, args:dict):
"stdev": statistics.stdev(duration_list) if len(duration_list) > 1 else 0}
for step_name, duration_list in stats.items()}
if args["model_cache"]: # FIL-First Inference Latency
agg_stats['time_to_first_inference_result (ms)']={'avg': (agg_stats['time_to_inference']['avg']+agg_stats['first_inference']['avg']) / 10 **3}
agg_stats['time_to_first_inference_result']={'avg': str((agg_stats['time_to_inference']['avg']+agg_stats['first_inference']['avg']) / 10 **3)+'ms'}
else: #FEIL-First Ever Inference Latency
agg_stats['time_to_first_inference_result (s)']={'avg': (agg_stats['time_to_inference']['avg']+agg_stats['first_inference']['avg']) / 10 **6}
agg_stats['time_to_first_inference_result']={'avg': str((agg_stats['time_to_inference']['avg']+agg_stats['first_inference']['avg']) / 10 **6)+'s'}
return agg_stats