From a8d126b2a2edbd9ede59b656215f9ecbc3048910 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 19:32:41 -0400 Subject: [PATCH] pt dpp some ignores --- pytorch_lightning/root_module/memory.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/root_module/memory.py b/pytorch_lightning/root_module/memory.py index 44f45f35..389c8680 100644 --- a/pytorch_lightning/root_module/memory.py +++ b/pytorch_lightning/root_module/memory.py @@ -45,7 +45,7 @@ class ModelSummary(object): for i in range(1, len(mods)): m = mods[i] - if type(input_) is list or type(input_) is tuple: + if type(input_) is list or type(input_) is tuple: # pragma: no cover out = m(*input_) else: out = m(input_) @@ -62,7 +62,7 @@ class ModelSummary(object): in_sizes.append(in_size) - if type(out) is tuple or type(out) is list: + if type(out) is tuple or type(out) is list: # pragma: no cover out_size = np.asarray([x.size() for x in out]) else: out_size = np.array(out.size())