修改PyTorch预训练模型的下载路径

修改PyTorch预训练模型的下载路径


torchvision.model下载网络模型时不能指定路径,可以在下载之前用以下代码指定下载路径

import os
os.environ['TORCH_HOME'] = 'DownloadPath'

你可能感兴趣的:(Python,Pytorch,pytorch,深度学习,人工智能)