Commit
·
3c1f9ba
1
Parent(s):
61e12ef
Update README.md
Browse files
README.md
CHANGED
|
@@ -3,3 +3,28 @@ license: other
|
|
| 3 |
license_name: radio
|
| 4 |
license_link: LICENSE
|
| 5 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
license_name: radio
|
| 4 |
license_link: LICENSE
|
| 5 |
---
|
| 6 |
+
|
| 7 |
+
# AM-RADIO: Reduce All Domains Into One
|
| 8 |
+
|
| 9 |
+
## Pretrained Models
|
| 10 |
+
|
| 11 |
+
Refer to `model_results.csv` for model versions and their metrics.
|
| 12 |
+
|
| 13 |
+
### HuggingFace Hub
|
| 14 |
+
|
| 15 |
+
_Coming Soon_
|
| 16 |
+
|
| 17 |
+
### TorchHub
|
| 18 |
+
|
| 19 |
+
```Python
|
| 20 |
+
import torch
|
| 21 |
+
|
| 22 |
+
# If you don't supply the `version` parameter, the latest ViT version will be returned.
|
| 23 |
+
model = torch.hub.load('NVlabs/RADIO', 'radio_model', version='radio_v1', progress=True)
|
| 24 |
+
|
| 25 |
+
print(model)
|
| 26 |
+
```
|
| 27 |
+
|
| 28 |
+
## Training
|
| 29 |
+
|
| 30 |
+
_Coming Soon_
|