first model commit

README.md

@@ -8,7 +8,7 @@ pipeline_tag: image-classification
 language:
 - en
 tags:
-- resnet
+- convnext
 - convolutional neural network
 - simpool
 - dino
@@ -16,9 +16,9 @@ tags:
 - deep learning
 ---
 
-# Self-supervised ResNet-50 model
+# Self-supervised ConvNeXt-S model
 
-ResNet-50 official model trained on ImageNet-1k for 100 epochs. Self-supervision with [DINO](https://arxiv.org/abs/2104.14294). Reproduced for ICCV 2023 [SimPool](https://arxiv.org/abs/2309.06891) paper.
+[ConvNeXt-S](https://arxiv.org/abs/2201.03545) official model trained on ImageNet-1k for 100 epochs. Self-supervision with [DINO](https://arxiv.org/abs/2104.14294). Reproduced for ICCV 2023 [SimPool](https://arxiv.org/abs/2309.06891) paper.
 
 SimPool is a simple attention-based pooling method at the end of network, released in this [repository](https://github.com/billpsomas/simpool/).
 Disclaimer: This model card is written by the author of SimPool, i.e. [Bill Psomas](http://users.ntua.gr/psomasbill/).
@@ -36,6 +36,16 @@ Disclaimer: This model card is written by the author of SimPool, i.e. [Bill Psom
 
 ## BibTeX entry and citation info
 
+```
+@inproceedings{liu2022convnet,
+  title={A convnet for the 2020s},
+  author={Liu, Zhuang and Mao, Hanzi and Wu, Chao-Yuan and Feichtenhofer, Christoph and Darrell, Trevor and Xie, Saining},
+  booktitle={Proceedings of the IEEE/CVF conference on computer vision and pattern recognition},
+  pages={11976--11986},
+  year={2022}
+}
+```
+
 ```
 @misc{psomas2023simpool,
       title={Keep It SimPool: Who Said Supervised Transformers Suffer from Attention Deficit?},