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 857f33015a
Add native PyTorch weights for MixNet-Medium with no SAME padding necessary. Remove unused block of code.
6 years ago
..
data Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl 6 years ago
loss Big re-org, working towards making pip/module as 'timm' 6 years ago
models Add native PyTorch weights for MixNet-Medium with no SAME padding necessary. Remove unused block of code. 6 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 6 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 6 years ago