Skip to content

Commit 06a85aa

Browse files
committed
typo fix
1 parent afdbb85 commit 06a85aa

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

examples/models/mobilenet_v2/model.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,8 +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),)
61-
60+
return input_batch
6261

6362
class MV2UntrainedModel(EagerModelBase):
6463
def __init__(self):

0 commit comments

Comments
 (0)