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 63e677d03b Merge branch 'master' of github.com:rwightman/pytorch-models 2019-04-10 14:55:54 -07:00
data Merge branch 'master' of github.com:rwightman/pytorch-models 2019-04-10 14:55:54 -07:00
loss Add smooth loss 2019-04-05 20:50:26 -07:00
models Merge branch 'master' of github.com:rwightman/pytorch-models 2019-04-10 14:55:54 -07:00
optim
scheduler
distributed_train.sh
inference.py Lots of refactoring and cleanup. 2019-04-10 14:53:34 -07:00
train.py Cleanup CheckpointSaver, add support for increasing or decreasing metric, switch to prec1 metric in train loop 2019-04-07 10:22:55 -07:00
utils.py Cleanup CheckpointSaver, add support for increasing or decreasing metric, switch to prec1 metric in train loop 2019-04-07 10:22:55 -07:00
validate.py Lots of refactoring and cleanup. 2019-04-10 14:53:34 -07:00