1. 30 Apr, 2019 1 commit
    • Bar's avatar
      Add ShuffleNet v2 (#849) · 7a4845a9
      Bar authored
      * 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
      7a4845a9
  2. 28 Mar, 2019 1 commit
    • Francisco Massa's avatar
      Add MobileNet V2 (#818) · a61803f0
      Francisco Massa authored
      * Add MobileNet V2
      
      * Remove redundant functions and make tests pass
      
      * Simplify a bit the implementation
      
      * Reuse ConvBNReLU more often
      
      * Remove input_size and minor changes
      
      * Py2 fix
      a61803f0
  3. 07 Mar, 2019 1 commit
    • Michael Kösel's avatar
      Add GoogLeNet (Inception v1) (#678) · a2093007
      Michael Kösel authored
      * 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
      a2093007
  4. 07 Oct, 2017 1 commit
  5. 20 Sep, 2017 1 commit
  6. 02 Jun, 2017 1 commit
    • Sasank Chilamkurthy's avatar
      Improve torchvision documentation (#179) · 432aa00d
      Sasank Chilamkurthy authored
      * Add documentation for transforms
      
      * document and remove unused imports in mnist.py
      
      * document lsun, mscoco datasets
      
      * rest of the datasets documented
      
      * Clean up the documentation in other functions
      
      * Add links for datasets
      
      * Add more documentation
      
      * pep8 fix
      432aa00d
  7. 31 May, 2017 1 commit
  8. 23 Mar, 2017 1 commit
  9. 16 Mar, 2017 1 commit
  10. 13 Mar, 2017 1 commit
  11. 10 Mar, 2017 1 commit
  12. 11 Feb, 2017 1 commit
    • Marat Dukhan's avatar
      SqueezeNet 1.0 and 1.1 models (#49) · d44273b4
      Marat Dukhan authored
      * Add SqueezeNet 1.0 and 1.1 models
      * Selectively avoid inplace in SqueezeNet
      * Use Glorot uniform initialization in SqueezeNet
      * Make all ReLU in SqueezeNet in-place
      * Add pretrained SqueezeNet 1.0 and 1.1
      * Minor fixes in SqueezeNet models
      d44273b4
  13. 17 Jan, 2017 1 commit
  14. 09 Jan, 2017 1 commit