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.
pytorch-image-models/timm
Ross Wightman 77e2e0c4e3
Add new auto-augmentation Tensorflow EfficientNet weights, incl B6 and B7 models. Validation scores still pending but looking good.
5 years ago
..
data Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl 5 years ago
loss Big re-org, working towards making pip/module as 'timm' 6 years ago
models Add new auto-augmentation Tensorflow EfficientNet weights, incl B6 and B7 models. Validation scores still pending but looking good. 5 years ago
optim Big re-org, working towards making pip/module as 'timm' 6 years ago
scheduler Make min-lr and cooldown-epochs cmdline args, change dash in color_jitter arg for consistency 5 years ago
__init__.py Add model registry and model listing fns, refactor model_factory/create_model fn 6 years ago
utils.py Add explicit half/fp16 support to loader and validation script 6 years ago
version.py Update version, results csv files, and move remaining dropbox weights to github 5 years ago