diff --git a/tb_plugins/profiling/tb_plugin/torch_tb_profiler/profiler/run_generator.py b/tb_plugins/profiling/tb_plugin/torch_tb_profiler/profiler/run_generator.py index 958f8ce354ae65add41da986da7f699f887f546a..e32f556503de840c4fe49627866c80ea33629922 100644 --- a/tb_plugins/profiling/tb_plugin/torch_tb_profiler/profiler/run_generator.py +++ b/tb_plugins/profiling/tb_plugin/torch_tb_profiler/profiler/run_generator.py @@ -434,9 +434,9 @@ class RunGenerator(object): temp = self.statistic_data[call_name] temp['Max'] = max(temp['Max'], call_duration) temp['Min'] = min(temp['Min'], call_duration) - temp['Total'] += call_duration + temp['Total'] = round(temp['Total'] + call_duration, 2) temp['Calls'] += 1 - temp['Average'] = temp['Total'] / temp['Calls'] + temp['Average'] = round(temp['Total'] / temp['Calls'], 2) else: self.statistic_data[call_name] = { 'Calls': 1,