fix(//tests/modules): Remove lenet test and rename generated

mobilenet_v2 module in hub.py Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Mar 31, 2020 · 4b58d3b · 4b58d3b
1 parent aef6003
commit 4b58d3b
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 17 deletions.
diff --git a/tests/modules/hub.py b/tests/modules/hub.py
@@ -10,7 +10,7 @@
           "inception": models.inception_v3(pretrained=True),
           "googlenet": models.googlenet(pretrained=True),
           "shufflenet": models.shufflenet_v2_x1_0(pretrained=True),
-          "mobilenet": models.mobilenet_v2(pretrained=True),
+          "mobilenet_v2": models.mobilenet_v2(pretrained=True),
           "resnext50_32x4d": models.resnext50_32x4d(pretrained=True),
           "wideresnet50_2": models.wide_resnet50_2(pretrained=True),
           "mnasnet": models.mnasnet1_0(pretrained=True),

diff --git a/tests/modules/test_compiled_modules.cpp b/tests/modules/test_compiled_modules.cpp
@@ -28,11 +28,9 @@ TEST_P(ModuleTests, CompiledModuleIsClose) {
 INSTANTIATE_TEST_SUITE_P(CompiledModuleForwardIsCloseSuite,
                          ModuleTests,
                          testing::Values(
-                             PathAndInSize({"tests/modules/lenet.jit.pt",
-                                            {{1,1,28,28}}}),
-                             PathAndInSize({"tests/modules/resnet18.jit.pt",
-                                             {{1,3,224,224}}}),
-                             PathAndInSize({"tests/modules/resnet50.jit.pt",
-                                            {{1,3,224,224}}}),
-                             PathAndInSize({"tests/modules/mobilenet_v2.jit.pt",
-                                            {{1,3,224,224}}})));
+                            PathAndInSize({"tests/modules/resnet18.jit.pt",
+                                          {{1,3,224,224}}}),
+                            PathAndInSize({"tests/modules/resnet50.jit.pt",
+                                          {{1,3,224,224}}}),
+                            PathAndInSize({"tests/modules/mobilenet_v2.jit.pt",
+                                          {{1,3,224,224}}})));
diff --git a/tests/modules/test_modules_as_engines.cpp b/tests/modules/test_modules_as_engines.cpp
@@ -19,11 +19,9 @@ TEST_P(ModuleTests, ModuleAsEngineIsClose) {
 INSTANTIATE_TEST_SUITE_P(ModuleAsEngineForwardIsCloseSuite,
                          ModuleTests,
                          testing::Values(
-                             PathAndInSize({"tests/modules/lenet.jit.pt",
-                                            {{1,1,28,28}}}),
-                             PathAndInSize({"tests/modules/resnet18.jit.pt",
-                                            {{1,3,224,224}}}),
-                             PathAndInSize({"tests/modules/resnet50.jit.pt",
-                                            {{1,3,224,224}}}),
-                             PathAndInSize({"tests/modules/mobilenet_v2.jit.pt",
-                                            {{1,3,224,224}}})));
+                            PathAndInSize({"tests/modules/resnet18.jit.pt",
+                                          {{1,3,224,224}}}),
+                            PathAndInSize({"tests/modules/resnet50.jit.pt",
+                                          {{1,3,224,224}}}),
+                            PathAndInSize({"tests/modules/mobilenet_v2.jit.pt",
+                                          {{1,3,224,224}}})));