From 41f91f75cbc0139cb3d635126df8351c0709aa22 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Sat, 7 Jan 2023 15:01:31 +0000 Subject: [PATCH] Fix CI --- pytorch_vision_snnmlp.md | 6 +++--- scripts/install_deps.sh | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/pytorch_vision_snnmlp.md b/pytorch_vision_snnmlp.md index d33d2296..2da87327 100644 --- 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 index 9001ab5a..8b511362 100755 --- 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