We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent afdbb85 commit 06a85aaCopy full SHA for 06a85aa
examples/models/mobilenet_v2/model.py
@@ -57,8 +57,7 @@ def get_example_inputs(self):
57
input_tensor = preprocess(input_image)
58
input_batch = input_tensor.unsqueeze(0)
59
input_batch = (input_batch,)
60
- return (torch.randn(tensor_size),)
61
-
+ return input_batch
62
63
class MV2UntrainedModel(EagerModelBase):
64
def __init__(self):
0 commit comments