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 6e9697eb9c Fix small bug in seresnet input size and eval transform handling of img size		6 years ago
data	Fix small bug in seresnet input size and eval transform handling of img size	6 years ago
loss	…
models	Fix small bug in seresnet input size and eval transform handling of img size	6 years ago
optim	Update a few comment, add some references	6 years ago
scheduler	Update a few comment, add some references	6 years ago
distributed_train.sh	…
inference.py	…
train.py	…
utils.py	Update a few comment, add some references	6 years ago
validate.py	Update a few comment, add some references	6 years ago