669 Commits (6306aa79e90cc4131cb75aff56f5a08ffe7ce64f)

 

All Branches   

Search

Author	SHA1	Message	Date
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	3b8f63084d	Clean checkpoint renames pth w/ SHA hash in a torch hub compatible way	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	db04677c94	Merge pull request #52 from rwightman/randaugment ... RandAugment and more	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	1f39d15f15	Allow float decay epochs arg for training, works out with step lr math	5 years ago
Ross Wightman	7b83e67f77	Pass drop connect arg through to EfficientNet models	5 years ago
Ross Wightman	31453b039e	Update Auto/RandAugment comments, README, more. ... * Add a weighted choice option for RandAugment * Adjust magnitude noise/std naming, config	5 years ago
Ross Wightman	4243f076f1	Adding RandAugment to AutoAugment impl, some tweaks to AA included	5 years ago
Ross Wightman	4748c6dff2	Fix non-prefetch variant of Mixup. Fixes #50	5 years ago
Ross Wightman	0d58c50fb1	Add TF RandAug weights for B5/B7 EfficientNet models.	5 years ago
Ross Wightman	62105ed4a0	Better differentiate sotabench WSL, SSL, and SWSL models via model_desc	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
Ross Wightman	187ecbafbe	Add support for loading args from yaml file (and saving them with each experiment)	5 years ago
rwightman	d3ba34ee7e	Fix Mobilenet V3 model name for sotabench. Minor res2net cleanup.	5 years ago
rwightman	b5a8bb52fd	Silly typos.	5 years ago
rwightman	7dc5d7a997	sotabench fail	5 years ago
Ross Wightman	96364fc336	Merge pull request #35 from rwightman/res2net_dla ... Add Res2net and DLA models. Update sotabench.	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	7fd0857bfe	Merge pull request #33 from rwightman/autoaugment ... Add AutoAugment ImageNet policies for training	5 years ago
Ross Wightman	4002c0d4ce	Fix AutoAugment abs translate calc	5 years ago
Ross Wightman	c06274e5a2	Add note on random selection of magnitude value	5 years ago
Ross Wightman	b750b76f67	More AutoAugment work. Ready to roll...	5 years ago
Ross Wightman	25d2088d9e	Working on auto-augment	5 years ago
Ross Wightman	aff194f42c	Merge pull request #32 from rwightman/opt ... More optimizer work	5 years ago
Ross Wightman	64966f61f7	Add Nvidia's NovogGrad impl from Jasper (cleaner/faster than current) and Apex Fused optimizers	5 years ago
Ross Wightman	3d9c8a6489	Add support for new AMP checkpointing support w/ amp.state_dict	5 years ago
Ross Wightman	ba3c97c3ad	Some Lookahead cleanup and fixes	5 years ago
Ross Wightman	5c6da1c55a	Experiment with sotabench model_description field.	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	f37e633e9b	Merge remote-tracking branch 'origin/re-exp' into opt	5 years ago
Ross Wightman	b06dce8d71	Bump version for next push to pypi	5 years ago
Ross Wightman	32a0ca03c8	Update csv results files	5 years ago
Ross Wightman	dc0630fd94	Merge pull request #30 from rwightman/mixnet_xl ... MixNet-XL	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	9816ca3ab4	Add EdgeTPU models to sotabench, remove some unicode chars from model names	5 years ago
Ross Wightman	9ec6824bab	Finally got around to adding EdgeTPU EfficientNet variant	5 years ago
Ross Wightman	daeaa113e2	Add initial sotabench attempt. Split create_transform out of create_loader. Update requirements.txt	5 years ago