Pytorch模型默认下载路径

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

评论

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

这个站点使用 Akismet 来减少垃圾评论。了解你的评论数据如何被处理