Fix hubconf example in README
Browse files
README.md
CHANGED
|
@@ -107,10 +107,10 @@ The pre-trained models can be accessed via [PyTorch Hub](https://pytorch.org/hub
|
|
| 107 |
```python
|
| 108 |
import torch
|
| 109 |
|
| 110 |
-
aim_600m = torch.hub.load("apple/ml-aim", "
|
| 111 |
-
aim_1b = torch.hub.load("apple/ml-aim", "
|
| 112 |
-
aim_3b = torch.hub.load("apple/ml-aim", "
|
| 113 |
-
aim_7b = torch.hub.load("apple/ml-aim", "
|
| 114 |
```
|
| 115 |
|
| 116 |
### Pre-trained backbones
|
|
|
|
| 107 |
```python
|
| 108 |
import torch
|
| 109 |
|
| 110 |
+
aim_600m = torch.hub.load("apple/ml-aim", "aim_600M")
|
| 111 |
+
aim_1b = torch.hub.load("apple/ml-aim", "aim_1B")
|
| 112 |
+
aim_3b = torch.hub.load("apple/ml-aim", "aim_3B")
|
| 113 |
+
aim_7b = torch.hub.load("apple/ml-aim", "aim_7B")
|
| 114 |
```
|
| 115 |
|
| 116 |
### Pre-trained backbones
|