From b16d1a3e924e71b1f70e01cab2b4cc873c673c55 Mon Sep 17 00:00:00 2001 From: giovannilopez9808 Date: Sat, 18 Mar 2023 21:09:06 -0600 Subject: [PATCH] [Fix] frame.append dreprecated --- torchstat/reporter.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/torchstat/reporter.py b/torchstat/reporter.py index c363b3a..46a8ef1 100644 --- a/torchstat/reporter.py +++ b/torchstat/reporter.py @@ -54,13 +54,16 @@ def report_format(collected_nodes): del df['duration'] # Add Total row - total_df = pd.Series([total_parameters_quantity, total_memory, - total_operation_quantity, total_flops, - total_duration, mread, mwrite, total_memrw], - index=['params', 'memory(MB)', 'MAdd', 'Flops', 'duration[%]', - 'MemRead(B)', 'MemWrite(B)', 'MemR+W(B)'], - name='total') - df = df.append(total_df) + total_df = pd.DataFrame([[total_parameters_quantity, total_memory, + total_operation_quantity, total_flops, + total_duration, mread, mwrite, total_memrw]], + columns=['params', 'memory(MB)', 'MAdd', 'Flops', 'duration[%]', + 'MemRead(B)', 'MemWrite(B)', 'MemR+W(B)'], + index=['total']) + df = pd.concat([ + df, + total_df + ]) df = df.fillna(' ') df['memory(MB)'] = df['memory(MB)'].apply(