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/models
Ross Wightman c099374771
Map pretrained checkpoint to cpu to avoid issue with some pretrained checkpoints still having CUDA tensors. Fixes #42
5 years ago
..
__init__.py Add Res2Net and DLA models w/ pretrained weights. Update sotabench. 5 years ago
adaptive_avgmax_pool.py Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py 5 years ago
conv2d_helpers.py Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl 5 years ago
densenet.py Fix typo in Densenet default resolutions 5 years ago
dla.py Add Res2Net and DLA models w/ pretrained weights. Update sotabench. 5 years ago
dpn.py Register dpn107 5 years ago
factory.py Add model registry and model listing fns, refactor model_factory/create_model fn 5 years ago
gen_efficientnet.py Add memory efficient Swish impl 5 years ago
gluon_resnet.py Model updates. Add my best ResNet50 weights top-1=78.47. Add some other torchvision weights. 5 years ago
gluon_xception.py A few more additions to Gluon Xception models to match interface of others. 5 years ago
helpers.py Map pretrained checkpoint to cpu to avoid issue with some pretrained checkpoints still having CUDA tensors. Fixes #42 5 years ago
inception_resnet_v2.py Add model registry and model listing fns, refactor model_factory/create_model fn 5 years ago
inception_v3.py Update version, results csv files, and move remaining dropbox weights to github 5 years ago
inception_v4.py Add model registry and model listing fns, refactor model_factory/create_model fn 5 years ago
median_pool.py Big re-org, working towards making pip/module as 'timm' 5 years ago
nasnet.py Add NASNet-Large model 5 years ago
pnasnet.py Add model registry and model listing fns, refactor model_factory/create_model fn 5 years ago
registry.py Model updates. Add my best ResNet50 weights top-1=78.47. Add some other torchvision weights. 5 years ago
res2net.py Fix Mobilenet V3 model name for sotabench. Minor res2net cleanup. 5 years ago
resnet.py Add Facebook Research Semi-Supervised and Semi-Weakly Supervised ResNet model weights. 5 years ago
senet.py Update version, results csv files, and move remaining dropbox weights to github 5 years ago
test_time_pool.py Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py 5 years ago
xception.py Add model registry and model listing fns, refactor model_factory/create_model fn 5 years ago