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

Added OrderedDict as an output datatype

Open SoumyadeepB opened this issue 3 years ago • 0 comments

Some pre-trained models like the deeplabv3_resnet50 returns an OrderedDict containing the semantic mask and the auxillary loss as two separate Tensors under the keys 'out' and 'aux'.

Usage of torchsummary with such models was not compatible since OrderedDict had not been considered as a potential output type.

Error thrown: 'collections.OrderedDict' object has no attribute 'size'

This issue has now been fixed.

SoumyadeepB avatar Mar 19 '22 02:03 SoumyadeepB