torchvision and timm
Datasets, transforms and pretrained models for PyTorch (torchvision); hundreds of state-of-the-art backbones (timm).
Intermediate
timm.create_model("convnext_tiny", pretrained=True, num_classes=5).
Spotted a mistake or want to add detail? Suggest an improvement — Content Contributors' edits are reviewed by editors before they go live.
Discussion (0)
Sign in or join free to ask questions and discuss this topic.