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 dfa9298b4e
Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl
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' 5 years ago
models Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl 5 years ago
optim Big re-org, working towards making pip/module as 'timm' 5 years ago
scheduler A number of tweaks to arguments, epoch handling, config 5 years ago
__init__.py Add model registry and model listing fns, refactor model_factory/create_model fn 5 years ago
utils.py Add explicit half/fp16 support to loader and validation script 5 years ago
version.py Add explicit half/fp16 support to loader and validation script 5 years ago