From 4b58d3b1c9b032b1fda77c721b6e15b66bab6836 Mon Sep 17 00:00:00 2001 From: Naren Dasan Date: Tue, 31 Mar 2020 12:21:08 -0700 Subject: [PATCH] fix(//tests/modules): Remove lenet test and rename generated mobilenet_v2 module in hub.py Signed-off-by: Naren Dasan Signed-off-by: Naren Dasan --- tests/modules/hub.py | 2 +- tests/modules/test_compiled_modules.cpp | 14 ++++++-------- tests/modules/test_modules_as_engines.cpp | 14 ++++++-------- 3 files changed, 13 insertions(+), 17 deletions(-) diff --git a/tests/modules/hub.py b/tests/modules/hub.py index 9498185cf0..8be1047837 100644 --- 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 index 5f817dc34d..199e1d81b5 100644 --- 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 index 3b7287c6c8..759ed136c9 100644 --- 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}}}))); \ No newline at end of file + 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}}}))); \ No newline at end of file