Update README.md
Browse files
README.md
CHANGED
@@ -79,7 +79,7 @@ predicted_class_idx = logits.argmax(-1).item()
|
|
79 |
print("Predicted class:", model.config.id2label[predicted_class_idx])
|
80 |
```
|
81 |
|
82 |
-
##
|
83 |
|
84 |
A publication on this work is currently in preparation. In the meantime, please refer to this model by using the following citation:
|
85 |
```
|
@@ -104,7 +104,7 @@ For the base ViT model used please refer to:
|
|
104 |
primaryClass={cs.CV}
|
105 |
}
|
106 |
```
|
107 |
-
Imagenet
|
108 |
|
109 |
```bibtex
|
110 |
@inproceedings{deng2009imagenet,
|
|
|
79 |
print("Predicted class:", model.config.id2label[predicted_class_idx])
|
80 |
```
|
81 |
|
82 |
+
## BibTeX & References
|
83 |
|
84 |
A publication on this work is currently in preparation. In the meantime, please refer to this model by using the following citation:
|
85 |
```
|
|
|
104 |
primaryClass={cs.CV}
|
105 |
}
|
106 |
```
|
107 |
+
For referring to Imagenet:
|
108 |
|
109 |
```bibtex
|
110 |
@inproceedings{deng2009imagenet,
|