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

Fix GoogleNet to GoogLeNet (#492)

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