例如:导入densenet121进行预训练
import torch
from torchvision import models
a = models.densenet121(pretrained=True)
##如果之前没有下载densenet121模型,运行代码后,代码会自动下载模型,并保存到
C:\Users\lvnianzu/.cache\torch\checkpoints
例如:导入densenet121进行预训练
import torch
from torchvision import models
a = models.densenet121(pretrained=True)
##如果之前没有下载densenet121模型,运行代码后,代码会自动下载模型,并保存到
C:\Users\lvnianzu/.cache\torch\checkpoints