本文重点
前面我们学习了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