From db9a8cfe788835940db3bd1775e0933018b8e7dc Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 19:24:58 -0400 Subject: [PATCH] ignoring dist parallel forward --- pytorch_lightning/root_module/memory.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: