The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more
Change clamp_min_ to clamp_(min=) as former doesn't work with DTensor / FSDP2 without custom registration
R
Ross Wightman committed
94b799bde77a78b19178fb4a3a798e69601704ae
Parent: 56467ad