Skip to content

Commit 5635941

Browse files
authored
Typo Fix for Default Example Inputs to MobilenetV2 Export (#13206)
### Summary [PR](#13019) forgot to change the return example input to use the default samoyed dog image when using real inputs. This PR fixes that typo. ### Test plan Now the example runners that use the `mv2` export contain the dog image in the pte by default.
1 parent afdbb85 commit 5635941

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/models/mobilenet_v2/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def get_example_inputs(self):
5757
input_tensor = preprocess(input_image)
5858
input_batch = input_tensor.unsqueeze(0)
5959
input_batch = (input_batch,)
60-
return (torch.randn(tensor_size),)
60+
return input_batch
6161

6262

6363
class MV2UntrainedModel(EagerModelBase):

0 commit comments

Comments
 (0)