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
|