Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Vasilis Vryniotis's avatar
    Adding more ConvNeXt variants + Speed optimizations (#5253) · 82929ae1
    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.
    82929ae1
models.rst 28.2 KB