From 51fdd3c638e5bfcd0fea9be494ac9a04d515a952 Mon Sep 17 00:00:00 2001 From: Namhyuk Ahn Date: Sun, 5 Aug 2018 13:03:29 +0900 Subject: [PATCH] Update torchsummaryX.py --- torchsummaryX/torchsummaryX.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torchsummaryX/torchsummaryX.py b/torchsummaryX/torchsummaryX.py index 694c229..7e0af2c 100644 --- a/torchsummaryX/torchsummaryX.py +++ b/torchsummaryX/torchsummaryX.py @@ -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