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 5c4991a088 Add PyTorch trained EfficientNet-ES weights from Andrew Lavin		4 years ago
..
__init__.py	Merge pull request #74 from rwightman/augmix-jsd	5 years ago
activations.py	Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo	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_layers.py	Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo	5 years ago
densenet.py	Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.	5 years ago
dla.py	Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.	5 years ago
dpn.py	Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.	5 years ago
efficientnet.py	Add PyTorch trained EfficientNet-ES weights from Andrew Lavin	4 years ago
efficientnet_blocks.py	Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo	5 years ago
efficientnet_builder.py	Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.	5 years ago
factory.py	Update EfficientNet comments, MobileNetV3 non-TF create fns, fix factory arg checks, bump PyTorch version req to 1.2	5 years ago
feature_hooks.py	Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo	5 years ago
gluon_resnet.py	Add ResNet deep tiered stem and model weights for seresnext26t_32x4d and seresnext26d_32x4d	5 years ago
gluon_xception.py	Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.	5 years ago
helpers.py	Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts	5 years ago
hrnet.py	Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.	5 years ago
inception_resnet_v2.py	InceptionResNetV2 torchscript compatible	5 years ago
inception_v3.py	Update version, results csv files, and move remaining dropbox weights to github	5 years ago
inception_v4.py	Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.	5 years ago
median_pool.py	Big re-org, working towards making pip/module as 'timm'	5 years ago
mobilenetv3.py	Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.	5 years ago
nasnet.py	Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.	5 years ago
pnasnet.py	Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.	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	ResNet / Res2Net additions:	5 years ago
resnet.py	New ResNet50 JSD + RandAugment weights	4 years ago
selecsls.py	Add selecsls60 weights	5 years ago
senet.py	Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.	5 years ago
split_batchnorm.py	Update AugMix, JSD, etc comments and references	5 years ago
test_time_pool.py	Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.	5 years ago
xception.py	Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.	5 years ago