（3）pokeman_采用resnet18作为预训练模型
发布时间
阅读量：
阅读量
1、模型引入与应用
from torchvision.models import resnet18
2、使用预训练模型
（1）pretrained=True表示启用预训练模型
（2）*list展开的含义
trained_model = resnet18(pretrained=True)
model = nn.Sequential(*list(trained_model.children())[:-1], #[b, 512, 1, 1]
Flatten(), # [b, 512, 1, 1] => [b, 512]
nn.Linear(512, 5)
).to(device)
（3）最终的输出层被替换为自定义的nn.Linear(512,class_numm)结构
3、代码实现与功能验证
import torch
from torch import optim, nn
import visdom全部评论 (0)
还没有任何评论哟~