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

don't print nans

Browse Source
This commit is contained in:
Mike Clark
2019-06-05 05:45:39 +00:00
committed by GitHub
parent 2db8410f43
commit 755069ccb3
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
@@ -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)