8 Commits (7d235c5a5fbb4eb0e8836303b6fe4222fdd02719)

Author	SHA1	Message	Date
Ross Wightman	372ad5fa0d	Significant model refactor and additions: ... * All models updated with revised foward_features / forward_head interface * Vision transformer and MLP based models consistently output sequence from forward_features (pooling or token selection considered part of 'head') * WIP param grouping interface to allow consistent grouping of parameters for layer-wise decay across all model types * Add gradient checkpointing support to a significant % of models, especially popular architectures * Formatting and interface consistency improvements across models * layer-wise LR decay impl part of optimizer factory w/ scale support in scheduler * Poolformer and Volo architectures added	3 years ago
Ross Wightman	abc9ba2544	Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.	3 years ago
Ross Wightman	bcec14d3b5	Bring EfficientNet SE layer in line with others, pull se_ratio outside of blocks. Allows swapping w/ other attn layers.	4 years ago
Ross Wightman	9611458e19	Throw in some FBNetV3 code I had lying around, some refactoring of SE reduction channel calcs for all EffNet archs.	4 years ago
Ross Wightman	c4f482a08b	EfficientNetV2 official impl w/ weights ported from TF. Cleanup/refactor of related EfficientNet classes and models.	4 years ago
Ross Wightman	d584e7f617	Support for huggingface hub via create_model and default_cfgs. ... * improve consistency of model creation helper fns * add comments to some of the model helpers * support passing external default_cfgs so they can be sourced from hub	4 years ago
Ross Wightman	1f799afcb3	Remove capital letters in model names	4 years ago
AFLALO, Jonathan Isaac	82c1fe958a	added hardcorenas model ... refactoring to support feature extraction	4 years ago