Multiple outputs AttributeError

sksq96 / pytorch-summary

Model summary in PyTorch similar to `model.summary()` in Keras

MIT License

3.98k stars 412 forks source link

Multiple outputs AttributeError #136

Open erichhhhho opened 4 years ago

erichhhhho commented 4 years ago

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

wtomin commented 4 years ago

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'.

TylerYep commented 4 years ago

See #124