[feature request] Add optional 'directory/path' parameter to save available pretrained models.
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 17.9k
- Forks
- 7.3k
- Avg merge
- 1d 15h
- Merged PRs (30d)
- 13
Description
I may be wrong but I could not find any simple way of specifying where the pre-trained models from torchvision.models are saved.
By default, they are saved in the ...\.torch\models\ directory.
It would be very helpful to have an optional parameter to set the download path for the pre-trained models. (something like model_dir parameter in torch.utils.model_zoo.load_url())
If there is an existing way to do this already, please correct me in the comments.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by inspecting how torchvision.models downloads pretrained models and compare it with torch.utils.model_zoo.load_url(), especially its model_dir parameter. Done means callers can optionally choose the download directory while the existing default remains unchanged; add or update coverage for saving a pretrained model there.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- computer-vision, machine-learning
- Issue type
- Feature
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100