设置PyTorch预训练模型默认下载目录

默认下载目录为:
C:\Users\用户名\.cache\torch

import os
os.environ['TORCH_HOME'] = 'E:/pytorch_pretrain_model'

发表回复

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

13 + 9 =