- 22 Oct, 2021 2 commits
-
-
Joao Gomes authored
* adding efficientnet to prototype * Removing _efficientnet_conf from prototype and main efficientnet classification models * fixing merge conflicts * fixing merge conflicts * fixing lint errors * fixing lint errors * fixing lint errors
-
Joao Gomes authored
* adding efficientnet to prototype * renaming _efficientnet_model in torchvision.model to be aligned with prototype implementation * adding specific cop sizes to ech model * adding specific cop and resize transforms * fixing lint errors * Update torchvision/prototype/models/efficientnet.py Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> * rename pretrained model versions Co-authored-by:
-
- 04 Oct, 2021 1 commit
-
-
Philip Meier authored
* add ufmt as code formatter * cleanup * quote ufmt requirement * split imports into more groups * regenerate circleci config * fix CI * clarify local testing utils section * use ufmt pre-commit hook * split relative imports into local category * Revert "split relative imports into local category" This reverts commit f2e224cde2008c56c9347c1f69746d39065cdd51. * pin black and usort dependencies * fix local test utils detection * fix ufmt rev * add reference utils to local category * fix usort config * remove custom categories sorting * Run pre-commit without fixing flake8 * got a double import in merge Co-authored-by:Nicolas Hug <nicolashug@fb.com>
-
- 30 Sep, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Moving _make_divisible to utils. * Replace the old ConvBNReLU and ConvBNActivation layers * Fix minor bug. * Moving SE layer to ops. * Adding deprecation warnings on old layers. * Apply changes to regnets.
-
- 21 Sep, 2021 1 commit
-
-
Kai Zhang authored
* allow custom activation in SqueezeExcitation * use ReLU as the default activation * make scale activation parameterizable Co-authored-by:
-
- 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.
-
