diff --git a/torchsummaryX/torchsummaryX.py b/torchsummaryX/torchsummaryX.py index 10e2eeb..fd4d76c 100644 --- a/torchsummaryX/torchsummaryX.py +++ b/torchsummaryX/torchsummaryX.py @@ -95,7 +95,7 @@ def summary(model, x, *args, **kwargs): df = df[['Kernel Shape', 'Output Shape', 'Params (K)', 'Mult-Adds (M)']] print("="*100) - print(df) + print(df.replace(np.nan, '-') print("="*100) print(df.sum()) print("="*100)