-
-
Notifications
You must be signed in to change notification settings - Fork 66
Open
Description
Can you please explain that only a few models have pretrained weights in the MetalHead.jl ? Thanks.
The following classification models are available, with or without pre-trained weights:
[AlexNet](https://pytorch.org/vision/stable/models/alexnet.html)
[ConvNeXt](https://pytorch.org/vision/stable/models/convnext.html)
[DenseNet](https://pytorch.org/vision/stable/models/densenet.html)
[EfficientNet](https://pytorch.org/vision/stable/models/efficientnet.html)
[EfficientNetV2](https://pytorch.org/vision/stable/models/efficientnetv2.html)
[GoogLeNet](https://pytorch.org/vision/stable/models/googlenet.html)
[Inception V3](https://pytorch.org/vision/stable/models/inception.html)
[MaxVit](https://pytorch.org/vision/stable/models/maxvit.html)
[MNASNet](https://pytorch.org/vision/stable/models/mnasnet.html)
[MobileNet V2](https://pytorch.org/vision/stable/models/mobilenetv2.html)
[MobileNet V3](https://pytorch.org/vision/stable/models/mobilenetv3.html)
[RegNet](https://pytorch.org/vision/stable/models/regnet.html)
[ResNet](https://pytorch.org/vision/stable/models/resnet.html)
[ResNeXt](https://pytorch.org/vision/stable/models/resnext.html)
[ShuffleNet V2](https://pytorch.org/vision/stable/models/shufflenetv2.html)
[SqueezeNet](https://pytorch.org/vision/stable/models/squeezenet.html)
[SwinTransformer](https://pytorch.org/vision/stable/models/swin_transformer.html)
[VGG](https://pytorch.org/vision/stable/models/vgg.html)
[VisionTransformer](https://pytorch.org/vision/stable/models/vision_transformer.html)
[Wide ResNet](https://pytorch.org/vision/stable/models/wide_resnet.html)
Metadata
Metadata
Assignees
Labels
No labels