bugfix: correct module path for torch.eye

gram-ai / capsule-networks

A PyTorch implementation of the NIPS 2017 paper "Dynamic Routing Between Capsules".

https://arxiv.org/abs/1710.09829

1.73k stars 316 forks source link

Closed topiaruss closed 6 years ago

topiaruss commented 6 years ago

As reported #17

Tested with pytorch v0.3.1

iwasaki-kenta commented 6 years ago

Closed w.r.t pull request #19.