- 09 May, 2022 1 commit
-
-
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
-
- 01 Feb, 2022 1 commit
-
-
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.
-
- 26 Aug, 2021 1 commit
-
-
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.
-
- 06 Aug, 2021 1 commit
-
-
Vincent Moens authored
using nn.init.trunc_normal_ instead of scipy.stats.truncnorm Co-authored-by:Vincent Moens <vmoens@fb.com>
-
- 22 Oct, 2019 1 commit
-
-
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
-