tests: add unit test

PaddlePaddle · Sep 24, 2021 · 137ff53 · 137ff53
1 parent d480d7d
commit 137ff53
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/paddle/tests/test_vision_models.py b/python/paddle/tests/test_vision_models.py
@@ -71,6 +71,9 @@ def test_resnet101(self):
     def test_resnet152(self):
         self.models_infer('resnet152')
 
+    def test_googlenet(self):
+        self.models_infer('GoogLeNet')
+
     def test_vgg16_num_classes(self):
         vgg16 = models.__dict__['vgg16'](pretrained=False, num_classes=10)