pytorch-summary icon indicating copy to clipboard operation
pytorch-summary copied to clipboard

Published 20 hours ago • verified publisher icon sksq96

Multiple outputs AttributeError

Open erichhhhho opened this issue 4 years ago • 2 comments

When certain layers have multiple outputs, there is an AttributeError: 'list' object has no attribute 'size'. Recommend adding feature to support multiple outputs.

erichhhhho avatar Jun 01 '20 06:06 erichhhhho

I got similar error. My model output is a dictionary consisting of the features and the output classification. I got an AttributeError: 'dcit' object has no attribute 'size'.

wtomin avatar Jun 04 '20 11:06 wtomin

See #124

TylerYep avatar Jun 06 '20 23:06 TylerYep