7 Commits (eb333d6641e2d68d63bddf731adb58909fc1b89c)

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	6c17d57a2c	Fix some attributions, add copyrights to some file docstrings	4 years ago
Ross Wightman	08016e839d	Cleanup FeatureInfo getters, add TF models sourced Xception41/65/71 weights	4 years ago
Ross Wightman	4e61c6a12d	Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript	4 years ago
Ross Wightman	6eec3fb4a4	Move FeatureHooks into features.py, switch EfficientNet, MobileNetV3 to use build model helper	4 years ago
Ross Wightman	3b9004bef9	Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces	4 years ago
Ross Wightman	d23a2697d0	Working on feature extraction, interfaces refined, a number of models working, some in progress.	4 years ago