加载resnet18的代码

from torchvision.models import resnet18
net = resnet18(pretrained=True)

猜你喜欢

转载自blog.csdn.net/huatianxue/article/details/109482815
今日推荐