pytorch-image-models/timm/models
Ross Wightman 4defbbbaa8 Fix module name mistake, start layers sub-package 2020-02-09 12:44:26 -08:00
..
layers Fix module name mistake, start layers sub-package 2020-02-09 12:44:26 -08:00
__init__.py Merge pull request #74 from rwightman/augmix-jsd 2020-01-11 12:04:29 -08:00
activations.py Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo 2019-11-28 17:04:35 -08:00
adaptive_avgmax_pool.py
conv2d_layers.py Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo 2019-11-28 17:04:35 -08:00
densenet.py Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment. 2019-11-29 17:56:36 -08:00
dla.py Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment. 2019-11-29 17:56:36 -08:00
dpn.py Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment. 2019-11-29 17:56:36 -08:00
efficientnet.py Add PyTorch trained EfficientNet-ES weights from Andrew Lavin 2020-02-06 12:53:55 -08:00
efficientnet_blocks.py Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo 2019-11-28 17:04:35 -08:00
efficientnet_builder.py Remove unused default_init for EfficientNets, experimenting with fanout calc for #84 2020-02-09 11:33:32 -08:00
factory.py Update EfficientNet comments, MobileNetV3 non-TF create fns, fix factory arg checks, bump PyTorch version req to 1.2 2019-11-28 17:43:00 -08:00
feature_hooks.py Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo 2019-11-28 17:04:35 -08:00
gluon_resnet.py Add ResNet deep tiered stem and model weights for seresnext26t_32x4d and seresnext26d_32x4d 2019-12-28 11:43:50 -08:00
gluon_xception.py Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment. 2019-11-29 17:56:36 -08:00
helpers.py Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts 2020-01-03 14:57:46 -08:00
hrnet.py Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment. 2019-11-29 17:56:36 -08:00
inception_resnet_v2.py InceptionResNetV2 torchscript compatible 2020-01-01 17:13:37 -08:00
inception_v3.py Update version, results csv files, and move remaining dropbox weights to github 2019-07-25 16:54:44 -07:00
inception_v4.py Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment. 2019-11-29 17:56:36 -08:00
median_pool.py
mobilenetv3.py Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment. 2019-11-29 17:56:36 -08:00
nasnet.py Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment. 2019-11-29 17:56:36 -08:00
pnasnet.py Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment. 2019-11-29 17:56:36 -08:00
registry.py
res2net.py ResNet / Res2Net additions: 2020-01-01 17:15:56 -08:00
resnet.py Fix module name mistake, start layers sub-package 2020-02-09 12:44:26 -08:00
selecsls.py Add selecsls60 weights 2019-12-31 16:49:04 -08:00
senet.py Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment. 2019-11-29 17:56:36 -08:00
split_batchnorm.py Update AugMix, JSD, etc comments and references 2020-01-11 12:02:05 -08:00
test_time_pool.py Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment. 2019-11-29 17:56:36 -08:00
xception.py Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment. 2019-11-29 17:56:36 -08:00