本文共 158 字,大约阅读时间需要 1 分钟。
网上看到有人暴力修改代码实现将下载的hub或model zoo的保存目录修改为自己喜欢的位置。
本质上pytorch已经提供环境变量重新定义模型下载后的保存目录:
import osos.environ["TORCH_HOME"]="E:\\pth2onnx\\models" #你要的保存目录,修改环境变量即可
转载地址:http://nvwl.baihongyu.com/