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
Add --fast-norm arg to benchmark.py, train.py, validate.py
R
Ross Wightman committed
ff6a919cf5f0a325236cf57c07548f779123173f
Parent: 769ab4b