wassname/torchsummaryX
1
0
Fork 0
mirror of https://github.com/wassname/torchsummaryX.git synced 2026-06-27 16:32:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

max rows to 600

Browse Source
This commit is contained in:
Mike Clark
2019-07-07 03:38:56 +00:00
committed by GitHub
parent a903488e39
commit f8896c6250
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
torchsummaryX/torchsummaryX.py
+1 -1
View File
@@ -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)