pytorch 日常操作

1 利用现成模型搭建迁移学习模型

import torchvision
import torch.nn as nn
seq = torchvision.models.resnet18(pretrained=True)
# 取出现有模型的layer,
seq = nn.Sequential(*list(seq.children())[:-2])
# 自定义需要的层
seq.add_module('maxPool2d', nn.AdaptiveMaxPool2d((1, 1)))

你可能感兴趣的:(pytorch)