search

cezannec / capsule_net_pytorch

Readable implementation of a Capsule Network as described in "Dynamic Routing Between Capsules" [Hinton et. al.]
MIT License
370 stars 130 forks source link

fixed deprecation warning in class CapsuleLoss #2

Closed datamove closed 5 years ago

datamove commented 5 years ago

in nn.MSELoss - now using reduction='sum' instead of size_average='False' as suggested in the doc page. https://pytorch.org/docs/stable/nn.html#mseloss

cezannec commented 5 years ago

Thank you! Happy to incorporate this update