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 824f42e75e
Forgot to include Tanh scheduler
6 years ago
models Cleanup tranforms, add custom schedulers, tweak senet34 model 6 years ago
optim Initial commit, puting some ol pieces together 6 years ago
scheduler Forgot to include Tanh scheduler 6 years ago
dataset.py Cleanup tranforms, add custom schedulers, tweak senet34 model 6 years ago
inference.py Initial commit, puting some ol pieces together 6 years ago
train.py Minor updates 6 years ago
utils.py Minor updates 6 years ago
validate.py Initial commit, puting some ol pieces together 6 years ago