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. 2019-07-30 18:31:02 -07:00
..
data Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl 2019-07-25 11:42:01 -07:00
loss Big re-org, working towards making pip/module as 'timm' 2019-06-19 17:20:51 -07:00
models Add new auto-augmentation Tensorflow EfficientNet weights, incl B6 and B7 models. Validation scores still pending but looking good. 2019-07-30 18:31:02 -07:00
optim Big re-org, working towards making pip/module as 'timm' 2019-06-19 17:20:51 -07:00
scheduler Make min-lr and cooldown-epochs cmdline args, change dash in color_jitter arg for consistency 2019-07-26 09:35:31 -07:00
__init__.py Add model registry and model listing fns, refactor model_factory/create_model fn 2019-06-23 18:22:16 -07:00
utils.py Add explicit half/fp16 support to loader and validation script 2019-07-05 13:52:25 -07:00
version.py Update version, results csv files, and move remaining dropbox weights to github 2019-07-25 16:54:44 -07:00