378 Commits (0004f37d25d59622ae82b3b9cc3583ddacff97c9)

 

All Branches   

Search

Author	SHA1	Message	Date
Ross Wightman	0004f37d25	Some TResNet cleanup. ... * allow use of global pool arg, test-time-pooling * clean checkpoints to just contain state dict, add 448 res checkpoints * support DataParallel via lazy filter creation for JIT Downsample * some minor formatting (mostly alignment) preferences	4 years ago
Ross Wightman	64fe37d008	Remove old mean/std helper, rely fully on cmd line or default_cfg now. Fixes #126	4 years ago
Ross Wightman	02a30411ad	Replace fp16 with amp support for validate.py script	4 years ago
Ross Wightman	e6f24e5578	Add 'momentum' optimizer (SGD w/o nesterov) for stable EfficientDet training defaults	4 years ago
Ross Wightman	ebf82b84ac	Merge pull request #122 from mrT23/master ... TResNet models	4 years ago
Ross Wightman	e15f979457	Merge pull request #123 from aclex/mobilenetv3_fix_feature_extraction ... Merge changes in feature extraction interface to MobileNetV3	4 years ago
Alexey Chernov	bdb165a8a4	Merge changes in feature extraction interface to MobileNetV3 ... Experimental feature extraction interface seems to be changed a little bit with the most up to date version apparently found in EfficientNet class. Here these changes are added to MobileNetV3 class to make it support it and work again, too.	4 years ago
talrid	8a63c1add8	finalizing	4 years ago
talrid	6209146738	TResNet models	4 years ago
mrT23	98a7403ed4	Merge pull request #1 from rwightman/master ... merge master	4 years ago
Ross Wightman	13cf68850b	Remove poorly named metrics from torch imagenet example origins. Use top1/top5 in csv output for consistency with existing validation results files, acc elsewhere. Fixes #111	4 years ago
Ross Wightman	56608c9070	Bump version for pypi	4 years ago
Ross Wightman	0ebcd2eca1	No default img size for inference script. Fixes #109	4 years ago
Ross Wightman	1a8f5900ab	Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.	4 years ago
Ross Wightman	e01ccb88ce	Merge pull request #117 from VRandme/typo_eca ... minor PR to fix typos found in #116 and resnet.py	4 years ago
Chris Ha	06a50a94a8	Fix minor typos in create_attn.py and resnet.py ... 'eca'->'ceca' and doest not-> does not	4 years ago
Ross Wightman	c99a5abed4	Merge pull request #115 from rwightman/mobilenetv2-experiment ... MobileNet-V2 experiments	4 years ago
Ross Wightman	e34074b4da	Add final weights for MobileNet-V2 experiments	4 years ago
Ross Wightman	a6a5565de7	Fiddling...	4 years ago
Ross Wightman	5a16c533ff	Add better resnext50_32x4d weights trained by andravin	4 years ago
Ross Wightman	bc998cad91	Experimenting with some MobileNetV2 variations to compare against EfficientNet-Lite	4 years ago
Ross Wightman	71b5cd67da	Merge pull request #105 from rwightman/efficientnet-lite ... EfficientNet-Lite	4 years ago
Ross Wightman	ef30e7f2b6	Update sotabench with EffNet Lite models	4 years ago
Ross Wightman	3406e582cf	Add EfficientNet-Lite results, update README	4 years ago
Ross Wightman	bd05258f7b	EfficientNet-Lite model added w/ converted checkpoints, validation in progress...	4 years ago
Ross Wightman	7deacf5477	Merge pull request #99 from andravin/save-last ... Modified save_checkpoint to always save last checkpoint. Fixes #98.	4 years ago
Andrew Lavin	d92cc4da54	Modified save_checkpoint to always save last checkpoint. Fixes #98.	4 years ago
Ross Wightman	56e2ac3a6d	Merge pull request #94 from rwightman/lr_noise ... Learning rate noise, MobileNetV3 weights, and activate MobileNetV3/EfficientNet weight init change	4 years ago
Ross Wightman	c16f25ced2	Add MobileNetV3 Large weights, results, update README and sotabench for merge	4 years ago
Ross Wightman	c60069c1eb	Annotate types on drop fns to avoid torchscript error	4 years ago
Ross Wightman	9fee316752	Enable fixed fanout calc in EfficientNet/MobileNetV3 weight init by default. Fix #84	4 years ago
Ross Wightman	27b3680d49	Revamp LR noise, move logic to scheduler base. Fixup PlateauLRScheduler and add it as an option.	4 years ago
Ross Wightman	cc5a11abba	version bump for PyPi update	4 years ago
Ross Wightman	514b0938c4	Experimenting with per-epoch learning rate noise w/ step scheduler	4 years ago
Ross Wightman	d77f45a6f6	Forgot to add skresnet34 to sotabench	4 years ago
Ross Wightman	6620770665	Simpler approach to loading entrypoints in hubconf works properly	4 years ago
Ross Wightman	e0685dd415	Merge pull request #88 from rwightman/attention ... A lot of attention and much more	4 years ago
Ross Wightman	f1860ef3a5	Update README.md, check in hubconf experiment (fully dynamic)	4 years ago
Ross Wightman	43225d110c	Unify drop connect vs drop path under 'drop path' name, switch all EfficientNet/MobilenetV3 refs to 'drop_path'. Update factory to handle new drop args.	4 years ago
Ross Wightman	f1d5f8a6c4	Update comments for Selective Kernel and DropBlock/Path impl, add skresnet34 weights	4 years ago
Ross Wightman	569419b38d	Tweak some comments, add SKNet models with weights to sotabench, remove an unused branch	4 years ago
Ross Wightman	91e2b33d72	Update results with latest NoisyStudent models and SelectiveKernel models	4 years ago
Ross Wightman	53c47479c4	Batch validation batch size adjustment, tweak L2 crop pct	4 years ago
Ross Wightman	08553e16b3	Merge branch 'master' into attention	4 years ago
Ross Wightman	fa38f24967	Update SK network configs, add weights for skresnet8 and skresnext50	4 years ago
Ross Wightman	f098fda2ca	Add map_location='cpu' to ModelEma resume, should improve #72	4 years ago
Ross Wightman	b9496995d8	Add L2-475 PyTorch preprocessing result, update sotabench for new weights	4 years ago
Ross Wightman	ba15ca47e8	Add ported EfficientNet-L2, B0-B7 NoisyStudent weights from TF TPU	4 years ago
Ross Wightman	5e6dbbaf30	Add CBAM for experimentation	4 years ago
Ross Wightman	d725991870	Remove debug print from ECA module	4 years ago