每天五分钟深度学习pytorch:使用Inception模块搭建GoogLeNet模型

本文重点

前面我们学习了Incetption模块，它的作用类似于vgg块对于VGG网络模型一样，本文我们使用Inception搭建GoogLeNet网络，如果使用卷积层开始从头开始搭建GoogleNet，那么这样看起来会很不清晰，我们使用已经封装好的Inception来搭建GoogLeNet网络

关键点

关键点在于Inception和Inception模块之间的通道对应，不然整个网络连接不起来，只要把握住这里的关键点，那么问题就可以解决了，下面我们来看一下代码

代码

文字版代码

import numpy as np
import torch
from torch import nn
from torch.autograd import Variable
from torchvision.datasets import CIFAR10
def conv_relu(in_channel, out_channel, kernel, str