Skip to content

Commit ed58d27

Browse files
authored
Fix CI (#314)
1 parent af3c1b9 commit ed58d27

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

pytorch_vision_snnmlp.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,11 @@ order: 10
1818

1919
```python
2020
import torch
21-
model = torch.hub.load('huawei-noah/SNNMLP', 'snnmlp_t', pretrained=True)
21+
model = torch.hub.load('huawei-noah/Efficient-AI-Backbones', 'snnmlp_t', pretrained=True)
2222
# or
23-
# model = torch.hub.load('huawei-noah/SNNMLP', 'snnmlp_s', pretrained=True)
23+
# model = torch.hub.load('huawei-noah/Efficient-AI-Backbones', 'snnmlp_s', pretrained=True)
2424
# or
25-
# model = torch.hub.load('huawei-noah/SNNMLP', 'snnmlp_b', pretrained=True)
25+
# model = torch.hub.load('huawei-noah/Efficient-AI-Backbones', 'snnmlp_b', pretrained=True)
2626
model.eval()
2727
```
2828

scripts/install_deps.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,4 +18,4 @@ pip install -q hydra-core opencv-python fvcore
1818
pip install -q --upgrade google-api-python-client
1919
pip install pytorchvideo
2020
pip install -q prefetch_generator # yolop
21-
pip install -q pretrainedmodels efficientnet_pytorch # hybridnets
21+
pip install -q pretrainedmodels efficientnet_pytorch webcolors # hybridnets

0 commit comments

Comments
 (0)