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 b5255960d9 Tweaking tanh scheduler, senet weight init (for BN), transform defaults 2019-02-13 23:11:09 -08:00
models Tweaking tanh scheduler, senet weight init (for BN), transform defaults 2019-02-13 23:11:09 -08:00
optim Initial commit, puting some ol pieces together 2019-02-01 22:07:34 -08:00
scheduler Tweaking tanh scheduler, senet weight init (for BN), transform defaults 2019-02-13 23:11:09 -08:00
dataset.py Cleanup tranforms, add custom schedulers, tweak senet34 model 2019-02-06 20:19:11 -08:00
inference.py Initial commit, puting some ol pieces together 2019-02-01 22:07:34 -08:00
train.py Tweaking tanh scheduler, senet weight init (for BN), transform defaults 2019-02-13 23:11:09 -08:00
utils.py Minor updates 2019-02-08 20:56:24 -08:00
validate.py Initial commit, puting some ol pieces together 2019-02-01 22:07:34 -08:00