diff --git a/pytorch_lightning/root_module/memory.py b/pytorch_lightning/root_module/memory.py index ffcf8572..44f45f35 100644 --- a/pytorch_lightning/root_module/memory.py +++ b/pytorch_lightning/root_module/memory.py @@ -50,7 +50,7 @@ class ModelSummary(object): else: out = m(input_) - if type(input_) is tuple or type(input_) is list: + if type(input_) is tuple or type(input_) is list: # pragma: no cover in_size = [] for x in input_: if type(x) is list: