PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more
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.
Go to file
Ross Wightman 9e296dbffb
Add seresnet26_32x4d cfg and weights + interpolation str->PIL enum fn
6 years ago
data Add seresnet26_32x4d cfg and weights + interpolation str->PIL enum fn 6 years ago
loss Add smooth loss 6 years ago
models Add seresnet26_32x4d cfg and weights + interpolation str->PIL enum fn 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
clean_checkpoint.py Add checkpoint clean script, add link to pretrained resnext50 weights 6 years ago
distributed_train.sh Fix distributed train script 6 years ago
inference.py Add per model crop pct, interpolation defaults, tie it all together 6 years ago
train.py Add per model crop pct, interpolation defaults, tie it all together 6 years ago
utils.py Update a few comment, add some references 6 years ago
validate.py Update a few comment, add some references 6 years ago