未验证 提交 1da9fdf3 编写于 作者: T Tingquan Gao 提交者: GitHub

Fix GoogleNet to GoogLeNet (#492)

上级 fdfce2d6
......@@ -95,9 +95,9 @@ class Inception(nn.Layer):
return cat
class GoogleNetDY(nn.Layer):
class GoogLeNetDY(nn.Layer):
def __init__(self, class_dim=1000):
super(GoogleNetDY, self).__init__()
super(GoogLeNetDY, self).__init__()
self._conv = ConvLayer(3, 64, 7, 2, name="conv1")
self._pool = MaxPool2D(kernel_size=3, stride=2)
self._conv_1 = ConvLayer(64, 64, 1, name="conv2_1x1")
......@@ -203,5 +203,5 @@ class GoogleNetDY(nn.Layer):
def GoogLeNet(**args):
model = GoogleNetDY(**args)
model = GoogLeNetDY(**args)
return model
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册