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.
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 |
