pytorch-image-models/timm/models
Ross Wightman 355aa152d5 Just leave it float for now, will look at fp16 later. Remove unused reference code. 2020-02-01 22:11:00 -08:00
..
__init__.py Move Selective Kernel blocks/convs to their own sknet.py file 2020-02-01 22:11:00 -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 Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py 2019-06-20 17:29:25 -07:00
conv2d_layers.py Add ConvBnAct layer to parallel integrated SelectKernelConv, add support for DropPath and DropBlock to ResNet base and SK blocks 2020-02-01 22:11:00 -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 newly added TF ported EfficientNet-B8 weights (RandAugment) 2020-02-01 18:01:14 -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 Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment. 2019-11-29 17:56:36 -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 Big re-org, working towards making pip/module as 'timm' 2019-06-19 17:20:51 -07:00
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
nn_ops.py Just leave it float for now, will look at fp16 later. Remove unused reference code. 2020-02-01 22:11:00 -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 Model updates. Add my best ResNet50 weights top-1=78.47. Add some other torchvision weights. 2019-06-29 15:50:33 -07:00
res2net.py Move SelectKernelConv to conv2d_layers and more 2020-02-01 22:11:00 -08:00
resnet.py Add ConvBnAct layer to parallel integrated SelectKernelConv, add support for DropPath and DropBlock to ResNet base and SK blocks 2020-02-01 22:11:00 -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
sknet.py Add ConvBnAct layer to parallel integrated SelectKernelConv, add support for DropPath and DropBlock to ResNet base and SK blocks 2020-02-01 22:11:00 -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