/HyperNetworks

PyTorch implementation of HyperNetworks (Ha et al., ICLR 2017) for ResNet (Residual Networks)

Primary LanguageJupyter NotebookGNU General Public License v3.0GPL-3.0

HyperNetworks

PyTorch implementation of HyperNetworks (Ha et al., ICLR 2017) for ResNet. The code is primarily for CIFAR-10 but it's super easy to use it for any other dataset. It's also very easy to use it for ResNet architectures of different depths.

(Please cite this repository if you use any of the code/diagrams here, Thanks! 😊😊)

To cite, please use the following BibTeX entry:

@article{gaurav2018hypernetsgithub,
  title={HyperNetworks(Github)},
  author={{Mittal}, G.},
  howpublished = {\url{https://github.com/g1910/HyperNetworks}},
  year={2018}
}

How to Run

python train.py

Working

model_diagram

model_diagram_simplified

forward_backward_pass