Add ConvNeXt-V2 and smaller model variants. (#1709)

2025-06-20 12:06:35 +00:00 · 2024-02-14 11:53:07 +02:00
parent b60064780d
commit 68f7655895
3 changed files with 214 additions and 51 deletions
--- a/candle-examples/examples/convnext/README.md
+++ b/candle-examples/examples/convnext/README.md
@ -1,6 +1,7 @@
 # candle-convnext

-[A ConvNet for the 2020s](https://arxiv.org/abs/2201.03545).
+[A ConvNet for the 2020s](https://arxiv.org/abs/2201.03545) and
+[ConvNeXt V2: Co-designing and Scaling ConvNets with Masked Autoencoders](https://arxiv.org/abs/2301.00808).

 This candle implementation uses a pre-trained ConvNeXt network for inference. The
 classification head has been trained on the ImageNet dataset and returns the