pretrained-models.pytorch icon indicating copy to clipboard operation
pretrained-models.pytorch copied to clipboard
verified publisher icon Cadene

Pretrained model paths

Open Tortoise17 opened this issue 4 years ago • 1 comments

When I import the pretrained models, by default, it downloads at first in

/home/user/.cache/torch/hub/ ..model*

Is it possible to set this to any other directory?

Please if you can guide me.

Tortoise17 avatar Sep 16 '21 12:09 Tortoise17

你可以按照readme中设置TORCH_HOME 但我更推荐你使用：

torch.hub.set_dir("your download folder")

lcycode avatar Nov 11 '22 12:11 lcycode