1. 09 May, 2022 1 commit
    • YosuaMichael's avatar
      Adding resnext101 64x4d model (#5935) · 4c02f103
      YosuaMichael authored
      * Add resnext101_64x4d model definition
      
      * Add test for resnext101_64x4d
      
      * Add resnext101_64x4d weight
      
      * Update checkpoint to use EMA weigth
      
      * Add quantization model signature for resnext101_64x4d
      
      * Fix class name and update accuracy using 1 gpu and batch_size=1
      
      * Apply ufmt
      
      * Update the quantized weight and accuracy that we still keep the training log
      
      * Add quantized expect file
      
      * Update docs and fix acc1
      
      * Add recipe for quantized to PR
      
      * Update models.rst
      4c02f103
  2. 01 Feb, 2022 1 commit
    • Vasilis Vryniotis's avatar
      Adding more ConvNeXt variants + Speed optimizations (#5253) · 82929ae1
      Vasilis Vryniotis authored
      * Refactor model builder
      
      * Add 3 more convnext variants.
      
      * Adding weights for convnext_small.
      
      * Fix minor bug.
      
      * Fix number of parameters for small model.
      
      * Adding weights for the base variant.
      
      * Adding weights for the large variant.
      
      * Simplify LayerNorm2d implementation.
      
      * Optimize speed of CNBlock.
      
      * Repackage weights.
      82929ae1
  3. 26 Aug, 2021 1 commit
    • Vasilis Vryniotis's avatar
      Add EfficientNet Architecture in TorchVision (#4293) · 37a9ee5b
      Vasilis Vryniotis authored
      * Adding code skeleton
      
      * Adding MBConvConfig.
      
      * Extend SqueezeExcitation to support custom min_value and activation.
      
      * Implement MBConv.
      
      * Replace stochastic_depth with operator.
      
      * Adding the rest of the EfficientNet implementation
      
      * Update torchvision/models/efficientnet.py
      
      * Replacing 1st activation of SE with SiLU.
      
      * Adding efficientnet_b3.
      
      * Replace mobilenetv3 assets with custom.
      
      * Switch to standard sigmoid and reconfiguring BN.
      
      * Reconfiguration of efficientnet.
      
      * Add repr
      
      * Add weights.
      
      * Update weights.
      
      * Adding B5-B7 weights.
      
      * Update docs and hubconf.
      
      * Fix doc link.
      
      * Fix typo on comment.
      37a9ee5b
  4. 06 Aug, 2021 1 commit
  5. 22 Oct, 2019 1 commit
    • fbbradheintz's avatar
      Correctness test implemented with old test architecture (#1511) · b60cb726
      fbbradheintz authored
      * correctness test implemented with old test architecture
      
      * reverted an unneeded change, ran flake8
      
      * moving to relative tolerance of 1 part in 10k for classification correctness checks
      
      * going down to 1 part in 1000 for correctness checks bc architecture differences
      
      * one percent relative tolerance
      b60cb726