pytorch
wayne980
这个作者很懒,什么都没留下…
展开
-
pytorch 更改预训练模型网络结构
一个继承nn.module的model它包含一个叫做children()的函数,这个函数可以用来提取出model每一层的网络结构,在此基础上进行修改即可,修改方法如下(去除后两层):resnet_layer = nn.Sequential(*list(model.children())[:-2])那么,接下来就可以构建我们的网络了:class Net(nn.Module): def...原创 2018-11-13 12:38:42 · 7853 阅读 · 1 评论 -
pytorch 预训练模型修改
# coding=UTF-8import torchvision.models as modelsimport torchimport torch.nn as nnimport mathimport torch.utils.model_zoo as model_zoo class CNN(nn.Module): def __init__(self, block, la...原创 2018-11-13 12:40:27 · 1863 阅读 · 0 评论