Added support for SVHN dataset, fixed torch.eye errors

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 315 forks source link

Closed theaaron123 closed 6 years ago

theaaron123 commented 6 years ago

Addresses Issue #11 by adding support for SVHN data set, also incorporates fix for torch.eye issue #18

iwasaki-kenta commented 6 years ago

Sounds good; care to also additionally fix up the torch.eye import for the original capsule_network.py?