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

Add dictionary support

Open Morpheus3000 opened this issue 6 years ago • 1 comments

Thanks for the great work on something essential that was sorely missing! I saw that it didn't have dictionary support. I personally use dictionary for outputs of my network, since they are hashed and is faster than using lists for multiple outputs, albeit at a slightly more memory consumption. On top of that, using dictionaries, makes for more readable code.

Morpheus3000 avatar Nov 20 '19 13:11 Morpheus3000

Thanks! Works perfect for me, although I have not tested it extensively.

CptPirx avatar Apr 20 '22 10:04 CptPirx