PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more
Go to file
Ross Wightman 8a33a6c90a Add checkpoint clean script, add link to pretrained resnext50 weights 2019-04-13 14:15:35 -07:00
data Fix small bug in seresnet input size and eval transform handling of img size 2019-04-13 10:06:43 -07:00
loss Add smooth loss 2019-04-05 20:50:26 -07:00
models Add checkpoint clean script, add link to pretrained resnext50 weights 2019-04-13 14:15:35 -07:00
optim Update a few comment, add some references 2019-04-12 23:16:49 -07:00
scheduler Update a few comment, add some references 2019-04-12 23:16:49 -07:00
clean_checkpoint.py Add checkpoint clean script, add link to pretrained resnext50 weights 2019-04-13 14:15:35 -07:00
distributed_train.sh
inference.py Add per model crop pct, interpolation defaults, tie it all together 2019-04-12 22:55:24 -07:00
train.py Add per model crop pct, interpolation defaults, tie it all together 2019-04-12 22:55:24 -07:00
utils.py Update a few comment, add some references 2019-04-12 23:16:49 -07:00
validate.py Update a few comment, add some references 2019-04-12 23:16:49 -07:00