82 Commits (d9a6a9d0af9f467ba4c8f3c163b4f5d3568e904d)

Author	SHA1	Message	Date
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	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	ec0dd4053a	Add updated RandAugment trained EfficientNet-B0 trained weights from @michaelklachko	5 years ago
Ross Wightman	40fea63ebe	Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts	5 years ago
Ross Wightman	53001dd292	ResNet / Res2Net additions: ... * ResNet torchscript compat * output_stride arg supported to limit network stride via dilations (support for dilation added to Res2Net) * allow activation layer to be changed via act_layer arg	5 years ago
Ross Wightman	f96b3e5e92	InceptionResNetV2 torchscript compatible	5 years ago
Ross Wightman	19d93fe454	Add selecsls60 weights	5 years ago
Ross Wightman	0062c15fb0	Update checkpoint url with modelzoo compatible ones.	5 years ago
Ross Wightman	b5315e66b5	Streamline SelecSLS model without breaking checkpoint compat. Move cfg handling out of model class. Update feature/pooling behaviour to match current.	5 years ago
Ross Wightman	d59a756c16	Run PyCharm autoformat on selecsls and change mix cap variables and model names to all lower	5 years ago
Ross Wightman	fb3a0f4bb8	Merge pull request #65 from mehtadushy/selecsls ... Incorporate SelecSLS Models	5 years ago
Ross Wightman	19fc205a4d	Update comments on the new SE-ResNeXt26 models	5 years ago
Ross Wightman	acc3ed2b8c	Add EfficientNet-B3 weights, trained from scratch with RA.	5 years ago
Dushyant Mehta	2404361f62	correct asset paths	5 years ago
Dushyant Mehta	31939311f6	Added SelecSLS models	5 years ago
rwightman	1f4498f217	Add ResNet deep tiered stem and model weights for seresnext26t_32x4d and seresnext26d_32x4d	5 years ago
Dushyant Mehta	32012a44fd	Added SelecSLS model	5 years ago
Ross Wightman	73b78459dc	Add update RandAugment MixNet-XL weights	5 years ago
Ross Wightman	ff421e5e09	New PyTorch trained EfficientNet-B2 weights with my RandAugment impl	5 years ago
Ross Wightman	3bef524f9c	Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.	5 years ago
Ross Wightman	6ca0828166	Update EfficientNet comments, MobileNetV3 non-TF create fns, fix factory arg checks, bump PyTorch version req to 1.2	5 years ago
Ross Wightman	eccbadca74	Update EfficientNet comments	5 years ago
Ross Wightman	902d32fb16	Renamed gen_efficientnet.py -> efficientnet.py	5 years ago
Ross Wightman	5a0a8de7e3	ResNet updates: ... * remove redundant GluonResNet model/blocks and use the code in ResNet for Gluon weights * change SEModules back to using AdaptiveAvgPool instead of mean, PyTorch issue long fixed	5 years ago
Ross Wightman	a39cc43374	Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo ... * Split MobileNetV3 and EfficientNet model files and put builder and blocks in own files (getting too large) * Finalize CondConv EfficientNet variant * Add the AdvProp weights files and B8 EfficientNet model * Refine the feature extraction module for EfficientNet and MobileNetV3	5 years ago
Ross Wightman	ad93347548	Initial HRNet classification model commit	5 years ago
Ross Wightman	2393708650	Missed stashing of out_indices in model	5 years ago
Ross Wightman	35e8f0c5e7	Fixup a few comments, add PyTorch version aware Flatten and finish as_sequential for GenEfficientNet	5 years ago
Ross Wightman	7ac6db4543	Missed activations.py	5 years ago
Ross Wightman	506df0e3d0	Add CondConv support for EfficientNet into WIP for GenEfficientNet Feature extraction setup	5 years ago
Ross Wightman	576d360f20	Bring in JIT version of optimized swish activation from gen_efficientnet as default (while working on feature extraction functionality here).	5 years ago
Ross Wightman	7b83e67f77	Pass drop connect arg through to EfficientNet models	5 years ago
Ross Wightman	0d58c50fb1	Add TF RandAug weights for B5/B7 EfficientNet models.	5 years ago
Ross Wightman	c099374771	Map pretrained checkpoint to cpu to avoid issue with some pretrained checkpoints still having CUDA tensors. Fixes #42	5 years ago
Ross Wightman	b93fcf0708	Add Facebook Research Semi-Supervised and Semi-Weakly Supervised ResNet model weights.	5 years ago
Ross Wightman	a9eb484835	Add memory efficient Swish impl	5 years ago
rwightman	d3ba34ee7e	Fix Mobilenet V3 model name for sotabench. Minor res2net cleanup.	5 years ago
Ross Wightman	2680ad14bb	Add Res2Net and DLA to README	5 years ago
rwightman	adbf770f16	Add Res2Net and DLA models w/ pretrained weights. Update sotabench.	5 years ago
Ross Wightman	aff194f42c	Merge pull request #32 from rwightman/opt ... More optimizer work	5 years ago
Ross Wightman	3d9c8a6489	Add support for new AMP checkpointing support w/ amp.state_dict	5 years ago
Ross Wightman	e9d2ec4d8e	Merge pull request #31 from rwightman/opt ... Optimizers and more	5 years ago
Ross Wightman	fac58f609a	Add RAdam, NovoGrad, Lookahead, and AdamW optimizers, a few ResNet tweaks and scheduler factory tweak. ... * Add some of the trendy new optimizers. Decent results but not clearly better than the standards. * Can create a None scheduler for constant LR * ResNet defaults to zero_init of last BN in residual * add resnet50d config	5 years ago
Ross Wightman	81875d52a6	Update sotabench model list, add Mean-Max pooling DPN variants, disable download progress	5 years ago
Ross Wightman	73fbd97ed4	Add weights for my MixNet-XL creation, include README updates for EdgeTPU models	5 years ago
Ross Wightman	51a2375b0c	Experimenting with a custom MixNet-XL and MixNet-XXL definition	5 years ago
Ross Wightman	9ec6824bab	Finally got around to adding EdgeTPU EfficientNet variant	5 years ago