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 d28ebf8b476d4de5f1557fdb59dbf74f95902339..958f8ce354ae65add41da986da7f699f887f546a 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 @@ -440,11 +440,10 @@ class RunGenerator(object): else: self.statistic_data[call_name] = { 'Calls': 1, - 'Core': call_type, - 'Max': call_duration, - 'Min': call_duration, 'Total': call_duration, - 'Average': call_duration + 'Min': call_duration, + 'Average': call_duration, + 'Max': call_duration } return display_row