310 Commits (4defbbbaa89049f1805951cda9db15f832c32695)

 

All Branches   

Search

Author	SHA1	Message	Date
Ross Wightman	4defbbbaa8	Fix module name mistake, start layers sub-package	4 years ago
Ross Wightman	7011cd0902	A little bit of ECA cleanup	4 years ago
Ross Wightman	46471df7b2	Merge pull request #82 from VRandme/eca ... ECA-Net Efficient Channel Attention	4 years ago
Ross Wightman	d0eb59ef46	Remove unused default_init for EfficientNets, experimenting with fanout calc for #84	4 years ago
Ross Wightman	cade829105	Add EfficientNet-ES to sotabench	4 years ago
Chris Ha	e6a762346a	Implement Adaptive Kernel selection ... When channel size is given, calculate adaptive kernel size according to original paper. Otherwise use the given kernel size(k_size), which defaults to 3	4 years ago
Chris Ha	6db087a1ff	Merge remote-tracking branch 'upstream/master' into eca	4 years ago
Chris Ha	904c618040	Update EcaModule.py ... Make pylint happy (commas, unused imports, missed imports)	4 years ago
Chris Ha	db91ba053b	EcaModule(CamelCase) ... CamelCased EcaModule. Renamed all instances of ecalayer to EcaModule. eca_module.py->EcaModule.py	4 years ago
Ross Wightman	5eb0e363a6	Update README.md	4 years ago
Ross Wightman	5c4991a088	Add PyTorch trained EfficientNet-ES weights from Andrew Lavin	4 years ago
Chris Ha	d04ff95eda	Merge branch 'master' into eca	4 years ago
Chris Ha	d63ae121d5	Clean up eca_module code ... functionally similar adjusted rwightman's version of reshaping and viewing. Use F.pad for circular eca version for cleaner code	4 years ago
Chris Ha	b0d45fd09c	Merge remote-tracking branch 'upstream/master'	4 years ago
Ross Wightman	d66819d1f3	Indentation mistake. Fixes #81	4 years ago
Ross Wightman	65cda1c63f	Merge pull request #83 from andravin/validation-batch-size-multiplier ... Added commandline argument validation-batch-size-multiplier.	4 years ago
Chris Ha	f87fcd7e88	Implement Eca modules ... implement ECA module by 1. adopting original eca_module.py into models folder 2. adding use_eca layer besides every instance of SE layer	4 years ago
Chris Ha	697e05cb3e	Merge remote-tracking branch 'upstream/master'	4 years ago
Andrew Lavin	b72013def8	Added commandline argument validation-batch-size-multiplier with default set to 1.	4 years ago
Ross Wightman	4808b3c32f	Bump version for PyPi update, fix few out of date README items/mistakes, add README updates for TF EfficientNet-B8 (RandAugment)	4 years ago
Ross Wightman	5c85389e26	Update README.md	4 years ago
Ross Wightman	820b73d280	Update README.md	4 years ago
Ross Wightman	82c0a2ff12	Update README.md ... Fix relative paths (I think)	4 years ago
Ross Wightman	1ffd2d0694	Add results/README.md	4 years ago
Chris Ha	89a32454e1	Merge remote-tracking branch 'upstream/master'	4 years ago
Ross Wightman	fd98fb33c5	Update sotabench with tf_efficientnet_b8 model	4 years ago
Ross Wightman	7c88356682	Add update results on ImageNet validation, ImageNetV2, ImageNet-A, and ImageNet-Sketch for all models	4 years ago
Ross Wightman	1daa303744	Add support to Dataset for class id mapping file, clean up a bit of old logic. Add results file arg for validation and update script.	4 years ago
Ross Wightman	91534522f9	Add newly added TF ported EfficientNet-B8 weights (RandAugment)	4 years ago
Ross Wightman	82dd60b33c	Rename results files for more clarity	4 years ago
Ross Wightman	5b7cc16ac9	Add warning about using sync-bn with zero initialized BN layers. Fixes #54	4 years ago
Ross Wightman	b18c19901e	Update README.md ... Typo	4 years ago
Ross Wightman	12dbc74742	New ResNet50 JSD + RandAugment weights	4 years ago
Chris Ha	0697ab183b	Update .gitignore	4 years ago
Ross Wightman	e39aae56b4	Update README.md ... Fix typo	5 years ago
Ross Wightman	7a17ee9c3b	Update README.md	5 years ago
Ross Wightman	e42503e1d0	Update sotabench	5 years ago
Ross Wightman	2f41905ba5	Update ResNet50 weights to AuxMix trained 78.994 top-1. A few commentes re 'tiered_narrow' tn variant.	5 years ago
Ross Wightman	cc0b1f4130	Add support to clean_checkpoint.py to remove aux_bn weights/biases from SplitBatchNorm	5 years ago
Ross Wightman	2a88412413	Update README.md	5 years ago
Ross Wightman	d9a6a9d0af	Merge pull request #74 from rwightman/augmix-jsd ... AugMix, JSD loss, SplitBatchNorm (Auxiliary BN), and more	5 years ago
Ross Wightman	3eb4a96eda	Update AugMix, JSD, etc comments and references	5 years ago
Ross Wightman	a28117ea46	Add tiered narrow ResNet (tn) and weights for seresnext26tn_32x4d	5 years ago
Ross Wightman	cfa951bceb	Update README.md	5 years ago
Ross Wightman	833066b540	A few minor things in SplitBN	5 years ago
Ross Wightman	7547119891	Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together	5 years ago
Ross Wightman	2e955cfd0c	Update RandomErasing with some improved arg names, tweak to aspect range	5 years ago
Ross Wightman	3cc0f91e23	Fix augmix variable name scope overlap, default non-blended mode	5 years ago
Ross Wightman	7622015258	Update README.md	5 years ago
Ross Wightman	ec0dd4053a	Add updated RandAugment trained EfficientNet-B0 trained weights from @michaelklachko	5 years ago