Pretrained weight tags for lucid.models.resnet_50_cls.
50-layer bottleneck ResNet (25.6 M params, top-1 76.13% / top-5 92.86%).
Attributes
IMAGENET1K_V1WeightEntryImageNet-1k V1 checkpoint (top-1 76.130% / top-5 92.862%),
sourced from
torchvision/ResNet50_Weights.IMAGENET1K_V1.DEFAULTWeightEntryAlias for
IMAGENET1K_V1.Notes
Reference: He, Zhang, Ren, Sun, "Deep Residual Learning for Image Recognition", CVPR 2016 (arXiv:1512.03385).
Examples
>>> from lucid.models import resnet_50_cls
>>> model = resnet_50_cls(pretrained=True).eval()