-
Notifications
You must be signed in to change notification settings - Fork 7.2k
Closed
Description
🚀 The feature
Once all model builders are updated to support multiple pretrained weights, we should replace the old builders with the new ones. This involves the following steps:
- Review the naming conventions for final time based on [FEEDBACK] Multi-weight support prototype API #5088. Also decide on the naming convention of _presets.py (
ImageNetEvalorClassificationEval?) - Port to
torchvision.modelsthe inplace code changes oftorchvision.prototype.models. Porting depends on having agreement on the deprecation of positional arguments across the entire TorchVision. If such agreement doesn't exist at the time of porting, we will need to remove the mandatory keyword arguments from the API prior landing it to the main area. - Delete
torchvision.prototype.modelsand their prototype tests. - Update the existing model tests (
pretrained=>weights) and ensure all work without modifying the expected files. - Update the reference scripts to work with the new API.
- Update the documentation of the new model builder methods.
- Update the tutorials and gallery with how to use the pretrained models and do preprocessing.
oke-aditya and zhiqwangoke-aditya and zhiqwang