!7208 update alexnet conv channel.

Merge pull request !7208 from linqingke/r1.0
pull/7208/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 65a21139b3

@ -30,9 +30,9 @@ class AlexNet(nn.Cell):
def __init__(self, num_classes=10, channel=3, phase='train', include_top=True):
super(AlexNet, self).__init__()
self.conv1 = conv(channel, 64, 11, stride=4, pad_mode="same", has_bias=True)
self.conv2 = conv(64, 192, 5, pad_mode="same", has_bias=True)
self.conv3 = conv(192, 384, 3, pad_mode="same", has_bias=True)
self.conv4 = conv(384, 256, 3, pad_mode="same", has_bias=True)
self.conv2 = conv(64, 128, 5, pad_mode="same", has_bias=True)
self.conv3 = conv(128, 192, 3, pad_mode="same", has_bias=True)
self.conv4 = conv(192, 256, 3, pad_mode="same", has_bias=True)
self.conv5 = conv(256, 256, 3, pad_mode="same", has_bias=True)
self.relu = P.ReLU()
self.max_pool2d = nn.MaxPool2d(kernel_size=3, stride=2, pad_mode='valid')

Loading…
Cancel
Save