Pytorch模型默认下载路径

15次阅读

共计 208 个字符,预计需要花费 1 分钟才能阅读完成。

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

正文完
 0