Notification: We use only the ImageNet-1K pre-trained weights, from TorchVision whenever possible (for the update on pre-trained weights in TorchVision, we always prefer the v0.7.0 weights). Most weights will be automatically downloaded, except:
*Need to provide download url in config.
**Need to provide pt/pth file path in config.
All Weights:
| Model | Filename | Link |
|---|---|---|
| ERFNet's Encoder** | erfnet_encoder_pretrained.pth.tar | Official Repo Weight |
| MobileNetV2* | mobilenet_v2-b0353104.pth | TorchVision Weight |
| MobileNetV3-Large** | mobilenet_v3_large-8738ca79.pth | TorchVision Weight |
| VGG-16 | vgg16-397923af.pth | TorchVision Weight |
| ResNet-18 | resnet18-5c106cde.pth | TorchVision Weight |
| ResNet-34 | resnet34-333f7ec4.pth | TorchVision Weight |
| ResNet-50 | resnet50-19c8e357.pth | TorchVision Weight |
| ResNet-101 | resnet101-5d3b4d8f.pth | TorchVision Weight |
| Swin-Tiny** | swin_tiny_patch4_window7_224.pth | Official Repo Weight |