第二次作业:卷积神经网络 part 2
【第一部分】问题总结 创建MobileNet V2时,第二个基本单元的步长与论文中不一致: MobileNet V2中使用了BatchNorm来优化网络,归一化来加速网络,HybridSN中添加后分类结果变差。 【第二部分】代码练习 使用MobileNet V1对CIFAR10进行分类 可分离卷积的实现: class Block(nn.Module): '''Depthwise conv + Pointwise conv''' def __init__(self, in_planes, out_planes, stride=1): super(Block, self).__init__() # Depthwise 卷积,3*3 的卷积核,分为 in_planes,即各层单独进行卷积 # group这个参数是用做分组卷积的,但是现在用的比较多的是groups = in_channel # 当groups = in_channel时,是在做的depth-wise conv的 self.conv1 = nn.Conv2d(in_planes, in_planes, kernel_size=3, stride=stride, padding=1, groups=in_planes, bias=False) # 在卷积神经网络的卷积层之后总会添加BatchNorm2d进行数据的归一化处理