michalk8 commited on
Commit
3d45e62
·
verified ·
1 Parent(s): a2df2af

Fix hubconf example in README

Browse files
Files changed (1) hide show
  1. README.md +4 -4
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", "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
 
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