pytorch-summary
pytorch-summary copied to clipboard
Add dictionary support
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.
Thanks! Works perfect for me, although I have not tested it extensively.