1161 Commits (0149ec30d7d39ae8ed65b95bd6ed601055a97f4c)

 

All Branches   

Search

Author	SHA1	Message	Date
Ross Wightman	ec0dd4053a	Add updated RandAugment trained EfficientNet-B0 trained weights from @michaelklachko	5 years ago
Ross Wightman	8662454bd7	Plural for averaging script.	5 years ago
Ross Wightman	40fea63ebe	Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts	5 years ago
Ross Wightman	4666cc9aed	Add --pin-mem arg to enable dataloader pin_memory (showing more benefit in some scenarios now), also add --torchscript arg to validate.py for testing models with jit.script	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	a7fe891a13	Update README.md ... Update readme with SelecSLS details.	5 years ago
Ross Wightman	e728d70831	Merge pull request #66 from rwightman/selecsls_updates ... SelecSLS updates	5 years ago
Ross Wightman	84ca3d1f4d	Add SelecSLS to sotabench list	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	0554b7942c	Update README.md	5 years ago
Ross Wightman	a4497afa21	Update sotabench.py	5 years ago
Ross Wightman	53f578e16f	Update README with B3 training details	5 years ago
Ross Wightman	19fc205a4d	Update comments on the new SE-ResNeXt26 models	5 years ago
Ross Wightman	8894a6bc08	Update sotabench.py with latest seresnext26 and efficientnet_b3 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	3afc2a4dc0	Some cleanup/improvements to AugMix impl: ... * make 'increasing' levels for Contrast, Color, Brightness, Saturation ops * remove recursion from faster blending mix * add config striing parsing for AugMix	5 years ago
Ross Wightman	232ab7fb12	Working on an implementation of AugMix with JensenShannonDivergence loss that's compatible with my AutoAugment and RandAugment impl	5 years ago
Ross Wightman	ff8688ca3d	Merge pull request #62 from rwightman/reduce-bn ... Distribute BatchNorm stats	5 years ago
Ross Wightman	5719b493ad	Missed update dist-bn logic for EMA model	5 years ago
Ross Wightman	a435ea1327	Change reduce_bn to distribute_bn, add ability to choose between broadcast and reduce (mean). Add crop_pct arg to allow selecting validation crop while training.	5 years ago
Ross Wightman	3bff2b21dc	Add support for keeping running bn stats the same across distributed training nodes before eval/save	5 years ago
Ross Wightman	0161de0127	Switch RandoErasing back to on GPU normal sampling	5 years ago
Ross Wightman	5d7af97981	Update README.md ... Update latest training hparam/command line with correct RandAugment config string (`noise` -> `mstd`)	5 years ago
Ross Wightman	3129bdb2c1	Update README.md	5 years ago
Ross Wightman	ff421e5e09	New PyTorch trained EfficientNet-B2 weights with my RandAugment impl	5 years ago
Ross Wightman	00b93407c7	Update results-all.csv with latest models/weights	5 years ago
Ross Wightman	5259dbcbb4	Update README.md with latest changes	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	3ceeedc441	Merge pull request #53 from rwightman/condconvs_and_features ... Major model merge (EfficientNet-CondConv, EfficientNet-AdvProp, TF MobileNetV3, HRNet, more)	5 years ago
Ross Wightman	7b3c235ccc	Update sotabench with new models	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	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