1. 19 Aug, 2022 1 commit
    • Sophia Zhi's avatar
      Add the S3D architecture to TorchVision (#6412) · 6de7021e
      Sophia Zhi authored
      
      
      * S3D initial commit
      
      * add model builder code and docstrings
      
      * change classifier submodule, populate weights enum
      
      * fix change of block args from List[List[int]] to ints
      
      * add VideoClassification to transforms
      
      * edit weights url for testing, add s3d to models.video init
      
      * norm_layer changes
      
      * norm_layer and args fix
      
      * Overwrite default dropout
      
      * Remove docs from internal submodules.
      
      * Fix tests
      
      * Adding documentation.
      
      * Link doc from main models.rst
      
      * Fix min_temporal_size
      
      * Adding crop/resize parameters in references script
      
      * Release weights.
      
      * Refactor dropout.
      
      * Adding the weights table in the doc
      Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
      Co-authored-by: default avatarVasilis Vryniotis <vvryniotis@fb.com>
      6de7021e
  2. 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
  3. 06 Aug, 2021 1 commit
  4. 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