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