Mention VGG in the readme. (#1573)

This commit is contained in:
Laurent Mazare
2024-01-12 09:59:29 +01:00
committed by GitHub
parent 6242276c09
commit 8e06bfb4fd
2 changed files with 8 additions and 3 deletions

View File

@ -1,7 +1,9 @@
# candle-repvgg
A candle implementation of inference using a pre-trained [repvgg](https://arxiv.org/abs/2101.03697).
This uses a classification head trained on the ImageNet dataset and returns the
[RepVGG: Making VGG-style ConvNets Great Again](https://arxiv.org/abs/2101.03697).
This candle implementation uses a pre-trained RepVGG network for inference. The
classification head has been trained on the ImageNet dataset and returns the
probabilities for the top-5 classes.
## Running an example