Skip to content

Commit 72fedc1

Browse files
mcr229facebook-github-bot
authored andcommitted
fix mv2/3 input shape
Differential Revision: D48761859 fbshipit-source-id: 917d02f3aa9db0898e584889693195aef74b61f1
1 parent 03429b6 commit 72fedc1

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

backends/xnnpack/test/models/mobilenet_v2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
class TestMobileNetV2(unittest.TestCase):
2121
mv2 = models.mobilenetv2.mobilenet_v2(weights=MobileNet_V2_Weights)
2222
mv2 = mv2.eval()
23-
model_inputs = (torch.ones(1, 3, 224, 244),)
23+
model_inputs = (torch.ones(1, 3, 224, 224),)
2424

2525
all_operators = {
2626
"executorch_exir_dialects_edge__ops_aten__native_batch_norm_legit_no_training_default",

backends/xnnpack/test/models/mobilenet_v3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
class TestMobileNetV3(unittest.TestCase):
2020
mv3 = models.mobilenetv3.mobilenet_v3_small(pretrained=True)
2121
mv3 = mv3.eval()
22-
model_inputs = (torch.ones(1, 3, 224, 244),)
22+
model_inputs = (torch.ones(1, 3, 224, 224),)
2323

2424
all_operators = {
2525
"executorch_exir_dialects_edge__ops_aten__native_batch_norm_legit_no_training_default",

0 commit comments

Comments
 (0)