nmhkahn/torchsummaryX

Support outputs in dict type?

Opened this issue · 0 comments

wangg12 commented

Currently the outputs should be list/tuple or a single Tensor, how to modify the code to support dict outputs?