Commit Graph

20 Commits (5a40c6a3c48672cf11a1995f9ebdb54f44903f09)

Author SHA1 Message Date
Ross Wightman 372ad5fa0d Significant model refactor and additions: 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 d584e7f617 Support for huggingface hub via create_model and default_cfgs. 4 years ago
Yusuke Uchida f6b56602f9 fix test_model_default_cfgs 4 years ago
Ross Wightman b1f1a54de9 More uniform treatment of classifiers across all models, reduce code duplication. 5 years ago
Ross Wightman 4e61c6a12d Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript 5 years ago
Ross Wightman 3b9004bef9 Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 5 years ago
Ross Wightman d23a2697d0 Working on feature extraction, interfaces refined, a number of models working, some in progress. 5 years ago
Ross Wightman eb7653614f Monster commit, activation refactor, VoVNet, norm_act improvements, more 5 years ago
Ross Wightman 0ea53cecc3 Merge branch 'master' into densenet_update_and_more 5 years ago
Ross Wightman afb6bd0669 Add backward and default_cfg tests and fix a few issues found. Fix 5 years ago
Vyacheslav Shults a7ebe09029 Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given. 5 years ago
Ross Wightman 13746a33fc Big move, layer modules and fn to timm/models/layers 5 years ago
Ross Wightman 19d93fe454 Add selecsls60 weights 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
Dushyant Mehta 2404361f62 correct asset paths 5 years ago
Dushyant Mehta 31939311f6 Added SelecSLS models 5 years ago
Dushyant Mehta 32012a44fd Added SelecSLS model 5 years ago