Fix CI (#314)

pytorch · Jan 9, 2023 · ed58d27 · ed58d27
1 parent af3c1b9
commit ed58d27
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pytorch_vision_snnmlp.md b/pytorch_vision_snnmlp.md
@@ -18,11 +18,11 @@ order: 10
 
 ```python
 import torch
-model = torch.hub.load('huawei-noah/SNNMLP', 'snnmlp_t', pretrained=True)
+model = torch.hub.load('huawei-noah/Efficient-AI-Backbones', 'snnmlp_t', pretrained=True)
 # or
-# model = torch.hub.load('huawei-noah/SNNMLP', 'snnmlp_s', pretrained=True)
+# model = torch.hub.load('huawei-noah/Efficient-AI-Backbones', 'snnmlp_s', pretrained=True)
 # or
-# model = torch.hub.load('huawei-noah/SNNMLP', 'snnmlp_b', pretrained=True)
+# model = torch.hub.load('huawei-noah/Efficient-AI-Backbones', 'snnmlp_b', pretrained=True)
 model.eval()
 ```
 

diff --git a/scripts/install_deps.sh b/scripts/install_deps.sh
@@ -18,4 +18,4 @@ pip install -q hydra-core opencv-python fvcore
 pip install -q --upgrade google-api-python-client
 pip install pytorchvideo
 pip install -q prefetch_generator  # yolop
-pip install -q pretrainedmodels efficientnet_pytorch  # hybridnets
+pip install -q pretrainedmodels efficientnet_pytorch webcolors # hybridnets