Skip to content

Commit

Permalink
fix(//tests/modules): Remove lenet test and rename generated
Browse files Browse the repository at this point in the history
mobilenet_v2 module in hub.py

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>
  • Loading branch information
narendasan committed Mar 31, 2020
1 parent aef6003 commit 4b58d3b
Show file tree
Hide file tree
Showing 3 changed files with 13 additions and 17 deletions.
2 changes: 1 addition & 1 deletion tests/modules/hub.py
Original file line number Diff line number Diff line change
Expand Up @@ -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),
Expand Down
14 changes: 6 additions & 8 deletions tests/modules/test_compiled_modules.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -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}}})));
14 changes: 6 additions & 8 deletions tests/modules/test_modules_as_engines.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -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}}})));

0 comments on commit 4b58d3b

Please sign in to comment.