From 1da9fdf379efcb05ed09e1834a66ab9e37b38d92 Mon Sep 17 00:00:00 2001
From: Tingquan Gao <gaotingquan@baidu.com>
Date: Thu, 17 Dec 2020 11:20:41 +0800
Subject: [PATCH] Fix GoogleNet to GoogLeNet (#492)

---
 ppcls/modeling/architectures/googlenet.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/ppcls/modeling/architectures/googlenet.py b/ppcls/modeling/architectures/googlenet.py
index e23f6cd5..534c6ff0 100644
--- a/ppcls/modeling/architectures/googlenet.py
+++ b/ppcls/modeling/architectures/googlenet.py
@@ -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
-- 
GitLab