GitLab

* add wide resnet

* add docstring for wide resnet

* update WRN-50-2 model

* add docs

* extend WRN docstring

* use pytorch storage for WRN

* fix rebase

* fix typo in docs

* Add initial mnasnet impl

* Remove all type hints, comply with PyTorch overall style

* Expose models

* Remove avgpool from features() and add separately

* Fix python3-only stuff, replace subclasses with functions

* fix __all__

* Fix typo

* Remove conditional dropout

* Make dropout functional

* Addressing @fmassa's feedback, round 1

* Replaced adaptive avgpool with mean on H and W to prevent collapsing the batch dimension

* Partially address feedback

* YAPF

* Removed redundant class vars

* Update urls to releases

* Add information to models.rst

* Replace init with kaiming_normal_ in fan-out mode

* Use load_state_dict_from_url

Also move weights from ShuffleNet to PyTorch bucket. Additionally, rename shufflenet to make it consistent with the other models

* Add ShuffleNet v2

Added 4 configurations: x0.5, x1, x1.5, x2
Add 2 pretrained models: x0.5, x1

* fix lint

* Change globalpool to torch.mean() call

* Add GoogLeNet (Inception v1)

* Fix missing padding

* Add missing ReLu to aux classifier

* Add Batch normalized version of GoogLeNet

* Use ceil_mode instead of padding and initialize weights using "xavier"

* Match BVLC GoogLeNet zero initialization of classifier

* Small cleanup

* use adaptive avg pool

* adjust network to match TensorFlow

* Update url of pre-trained model and add classification results on ImageNet

* Bugfix that improves performance by 1 point