网上看到有人暴力修改代码实现将下载的hub或model zoo的保存目录修改为自己喜欢的位置。
本质上pytorch已经提供环境变量重新定义模型下载后的保存目录:
import os
os.environ["TORCH_HOME"]="E:\\pth2onnx\\models" #你要的保存目录,修改环境变量即可
网上看到有人暴力修改代码实现将下载的hub或model zoo的保存目录修改为自己喜欢的位置。
本质上pytorch已经提供环境变量重新定义模型下载后的保存目录:
import os
os.environ["TORCH_HOME"]="E:\\pth2onnx\\models" #你要的保存目录,修改环境变量即可