mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
ignoring dist parallel forward
This commit is contained in:
1 parent
a3ad0e0ac1
commit
db9a8cfe78
1 file changed
+1
-1
@@ -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:
|
||||
|
||||
Reference in new issue
Block a user