- 25 Feb, 2022 1 commit
-
-
Aditya Oke authored
* Add ops.conv3d * Refactor for conv2d and 3d * Refactor * Fix bug * Addres review * Fix bug * nit fix * Fix flake * Final fix * remove documentation * fix linter * Update all the implementations to use new Conv * Small doc fix Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> Co-authored-by:
Joao Gomes <jdsgomes@fb.com>
-
- 01 Feb, 2022 2 commits
-
-
Vasilis Vryniotis authored
* Graduate ConvNeXt to main TorchVision area. * Linter and all var. * Renaming var and making named params mandatory.
-
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.
-
- 24 Jan, 2022 1 commit
-
-
Yoshitomo Matsubara authored
* renamed ImageNet weights * renamed COCO weights * renamed COCO with VOC labels weights * renamed Kinetics 400 weights * rename default with DEFAULT * update test * fix typos * update test * update test * update test * indent as w was weight_enum * revert * Adding back the capitalization test Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 20 Jan, 2022 1 commit
-
-
Vasilis Vryniotis authored
* Adding CNBlock and skeleton architecture * Completed implementation. * Adding model in prototypes. * Add test and minor refactor for JIT. * Fix mypy. * Fixing naming conventions. * Fixing tests. * Fix stochastic depth percentages. * Adding stochastic depth to tiny variant. * Minor refactoring and adding comments. * Adding weights. * Update default weights. * Fix transforms issue * Move convnext to prototype. * linter fix * fix docs * Addressing code review comments.
-