例如:导入densenet121进行预训练
import torch
from torchvision import models
a = models.densenet121(pretrained=True)
##如果之前没有下载densenet121模型,运行代码后,代码会自动下载模型,并保存到
C:\Users\lvnianzu/.cache\torch\checkpoints
pytorch自动下载模型时,模型保存位置
最新推荐文章于 2024-04-24 21:39:01 发布