From 755069ccb3b3ecc9d7bd6abbb46be581f5d1f8fe Mon Sep 17 00:00:00 2001 From: Mike Clark Date: Wed, 5 Jun 2019 05:45:39 +0000 Subject: [PATCH] don't print nans --- torchsummaryX/torchsummaryX.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)