From f8896c625005cdbfec85120aa9c9e8e52f0e1143 Mon Sep 17 00:00:00 2001 From: Mike Clark Date: Sun, 7 Jul 2019 03:38:56 +0000 Subject: [PATCH] max rows to 600 --- torchsummaryX/torchsummaryX.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchsummaryX/torchsummaryX.py b/torchsummaryX/torchsummaryX.py index 1273ef9..cdba769 100644 --- a/torchsummaryX/torchsummaryX.py +++ b/torchsummaryX/torchsummaryX.py @@ -102,7 +102,7 @@ def summary(model, x, *args, **kwargs): max_repr_width = max([len(row) for row in df.to_string().split("\n")]) - with pd.option_context("display.max_rows", 10, "display.max_columns", 5, 'display.float_format', pd.io.formats.format.EngFormatter(use_eng_prefix=True)): + with pd.option_context("display.max_rows", 600, "display.max_columns", 10, 'display.float_format', pd.io.formats.format.EngFormatter(use_eng_prefix=True)): print("="*max_repr_width) print(df.replace(np.nan, "-")) print("-"*max_repr_width)