ignoring dist parallel forward

This commit is contained in:
William Falcon
2019-07-24 19:24:58 -04:00
parent a3ad0e0ac1
commit db9a8cfe78
+1 -1
View File
@@ -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: