pytorch vgg16& resnet50简单的迁移学习
import torch
import torch.nn as nn
from torch.utils.data import dataset,dataloader,Dataset,DataLoader
from torchvision import transforms
from torchvision.datasets import ImageFolder
from torchvision.models import vgg16
vgg = vgg16(pretrained=True)
#固定...
原创
2020-10-26 10:07:23 ·
1020 阅读 ·
0 评论