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 4423963 commit dae9395Copy full SHA for dae9395
examples/models/mobilenet_v2/model.py
@@ -15,8 +15,8 @@
15
16
17
class MV2Model(EagerModelBase):
18
- def __init__(self, useRealInput=True):
19
- self.useRealInput = useRealInput
+ def __init__(self, use_real_input=True):
+ self.use_real_input = use_real_input
20
pass
21
22
def get_eager_model(self) -> torch.nn.Module:
@@ -28,7 +28,7 @@ def get_eager_model(self) -> torch.nn.Module:
28
def get_example_inputs(self):
29
tensor_size = (1, 3, 224, 224)
30
input_batch = (torch.randn(tensor_size),)
31
- if self.useRealInput:
+ if self.use_real_input:
32
logging.info("Loaded real input image dog.jpg")
33
import urllib
34
url, filename = ("https://github.com/pytorch/hub/raw/master/images/dog.jpg", "dog.jpg")
0 commit comments