Pytorch模型默认下载路径

Evan1995发布于 2019-09-02

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