Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Can the pretrained weights download folder be specified? #271

Open
mgreen-neura opened this issue May 6, 2024 · 1 comment
Open

Can the pretrained weights download folder be specified? #271

mgreen-neura opened this issue May 6, 2024 · 1 comment

Comments

@mgreen-neura
Copy link

Can I specify where the pre-trained weight files are saved?

By default, they are downloaded into the user's home directory but it then exceeds my home directory storage quota.
It would be fantastic if I could direct the download to the current working directory.

@giovannividetta
Copy link

giovannividetta commented May 14, 2024

Hi, @mgreen-neura!

From TractSeg FAQs:

Can I save the pretrained weights at a different location?

Per default the pretrained weights will be downloaded to and loaded from ~/.tractseg/. You can change this directory by adding weights_dir=/absolute/path_to_where/you_want_it to ~/.tractseg/config.txt in a new line (if the file does not exist yet you have to create it).
Normally the pretrained weights will automatically be downloaded in the background right when they are needed. In some cases you might want to download all of them at once. To do so you can simply run download_all_pretrained_weights and the weights will be download to ~/.tractseg/ or the location you specified in ~/.tractseg/config.txt.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants