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

Update torchsummaryX.py

Browse Source
This commit is contained in:
Namhyuk Ahn
2018-08-05 13:03:29 +09:00
committed by GitHub
parent a6ad5f983b
commit 51fdd3c638
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
torchsummaryX/torchsummaryX.py
+3 -3
View File
@@ -4,9 +4,9 @@ import torch.nn as nn
from collections import OrderedDict
def summary(model, x, *args, **kwargs):
"""Summary the given model.
Summarized information are output shape, kernel shape and
number of the parameters and operations (Mult-Adds)
"""Summarize the given input model.
Summarized information are 1) output shape, 2) kernel shape,
3) number of the parameters and 4) operations (Mult-Adds)
Args:
model (Module): Model to summarize