Ross Wightman
commited on
Commit
·
631bca3
1
Parent(s):
27728fc
Update README.md. Forgot to include hf_hub prefix in create_model use.
Browse files
README.md
CHANGED
@@ -32,7 +32,7 @@ import PIL
|
|
32 |
import timm
|
33 |
import torch
|
34 |
|
35 |
-
model = timm.create_model("timm/eca_nfnet_l0")
|
36 |
|
37 |
config = model.default_cfg
|
38 |
img_size = config["test_input_size"][-1] if "test_input_size" in config else config["input_size"][-1]
|
@@ -100,4 +100,4 @@ L0 model variant & pretraining:
|
|
100 |
doi = {10.5281/zenodo.4414861},
|
101 |
howpublished = {\url{https://github.com/rwightman/pytorch-image-models}}
|
102 |
}
|
103 |
-
```
|
|
|
32 |
import timm
|
33 |
import torch
|
34 |
|
35 |
+
model = timm.create_model("hf_hub:timm/eca_nfnet_l0")
|
36 |
|
37 |
config = model.default_cfg
|
38 |
img_size = config["test_input_size"][-1] if "test_input_size" in config else config["input_size"][-1]
|
|
|
100 |
doi = {10.5281/zenodo.4414861},
|
101 |
howpublished = {\url{https://github.com/rwightman/pytorch-image-models}}
|
102 |
}
|
103 |
+
```
|