- 19 Aug, 2022 1 commit
-
-
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:
Vasilis Vryniotis <datumbox@users.noreply.github.com> Co-authored-by:
Vasilis Vryniotis <vvryniotis@fb.com>
-
- 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
-