import torchvision
import os
# 设置TORCH_HOME环境变量,改变下载和缓存的默认路径
print(os.getcwd())
os.environ['TORCH_HOME'] = os.getcwd()
resnet_model = torchvision.models.resnet18(pretrained=True)
设置TORCH_HOME环境变量,改变下载和缓存的默认路径
最新推荐文章于 2025-02-26 13:15:31 发布
由于未提供博客具体内容,无法给出包含关键信息的摘要。
4051

被折叠的 条评论
为什么被折叠?



