wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

pt dpp some ignores

Browse Source
This commit is contained in:
William Falcon
2019-07-24 19:32:41 -04:00
parent 10c3266ed4
commit a8d126b2a2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/root_module/memory.py
+2 -2
View File
@@ -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())