Update README.md
Browse files
README.md
CHANGED
|
@@ -12,6 +12,14 @@ This model has been pushed to the Hub using the [PytorchModelHubMixin](https://h
|
|
| 12 |
- Library: ronnief1/RetFiner
|
| 13 |
- Docs: [More Information Needed]
|
| 14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
## Citation
|
| 16 |
|
| 17 |
Please cite the original paper if you use this model:
|
|
|
|
| 12 |
- Library: ronnief1/RetFiner
|
| 13 |
- Docs: [More Information Needed]
|
| 14 |
|
| 15 |
+
|
| 16 |
+
```python
|
| 17 |
+
import torch
|
| 18 |
+
import torch.nn as nn
|
| 19 |
+
from huggingface_hub import PyTorchModelHubMixin
|
| 20 |
+
from vit import VisionTransformer, vit_base_patch16, vit_large_patch16
|
| 21 |
+
|
| 22 |
+
|
| 23 |
## Citation
|
| 24 |
|
| 25 |
Please cite the original paper if you use this model:
|