Commit Graph

11 Commits (11704cc721197be84ed8a0a816dc54c1468f7cc0)

Author SHA1 Message Date
Ross Wightman c3be79a8b7 Rename _pretrained.py -> pretrained.py, not feasible to change the other files to same scheme without breaking uses
1 year ago
Ross Wightman 23b357f1df Add ported Tensorflow MaxVit weights. Add a few more CLIP ViT fine-tunes. Tweak some model tag names. Improve model tag name sorting. Update HF hub push config layout.
1 year ago
Ross Wightman ebb99a1f8d Initial multi-weight support, handled so old pretraing config handling co-exists with new tags.
2 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.
2 years ago
Ross Wightman 8880f696b6 Refactoring, cleanup, improved test coverage.
3 years ago
Ross Wightman d400f1dbdd Filter test models before creation for backward/torchscript tests
3 years ago
Ross Wightman 16f7aa9f54 Add default_cfg options for min_input_size / fixed_input_size, queries in model registry, and use for testing self-attn models
3 years ago
Ross Wightman 6c17d57a2c Fix some attributions, add copyrights to some file docstrings
4 years ago
Ross Wightman 9cc289f18c Exclude EfficientNet-L2 models from test
4 years ago
Ross Wightman b8762cc67d Model updates. Add my best ResNet50 weights top-1=78.47. Add some other torchvision weights.
5 years ago
Ross Wightman 171c0b88b6 Add model registry and model listing fns, refactor model_factory/create_model fn
5 years ago