PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more

mobilenet-v2vision-transformer-modelsresnetrandaugmentpytorchpretrained-weightspretrained-modelsnormalization-free-trainingnfnetsmobilenetv3augmixmobile-deep-learningmnasnetmixnetimagenet-classifierefficientnet-trainingefficientnetdual-path-networksdistributed-trainingcnn-classification

You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Ross Wightman 6f9a0c8ef2 Merge branch 'master' of github.com:rwightman/pytorch-models		6 years ago
data	Fixup Resnext, remove alternate shortcut types	6 years ago
models	Merge branch 'master' of github.com:rwightman/pytorch-models	6 years ago
optim	Add some Nvidia performance enhancements (prefetch loader, fast collate), and refactor some of training and model fact/transforms	6 years ago
scheduler	Resnext added, changes to bring it and seresnet in line with rest of models	6 years ago
inference.py	Fixup validate/inference script args, fix senet init for better test accuracy	6 years ago
train.py	Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models	6 years ago
utils.py	Minor updates	6 years ago
validate.py	Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models	6 years ago