之前的版本好像是下载到~/.torch/models/xxx
,最近我看下载位置发生了变化,转到了~/.cache/torch/checkpoints
,下载的时候使用的函数load_state_dict_from_url
在github的地址为link。
可以看到,通过设置全局变量export TORCH_HOME=/xx/xxx
即可修改下载的默认路径,修改后的路径为$TORCH_HOME/checkpoints/xxx
Pytorch模型默认下载路径
作者:
在
作者:
在
之前的版本好像是下载到~/.torch/models/xxx
,最近我看下载位置发生了变化,转到了~/.cache/torch/checkpoints
,下载的时候使用的函数load_state_dict_from_url
在github的地址为link。
可以看到,通过设置全局变量export TORCH_HOME=/xx/xxx
即可修改下载的默认路径,修改后的路径为$TORCH_HOME/checkpoints/xxx
发表回复