Skip to content

Commit dae9395

Browse files
committed
Naming convention fix
1 parent 4423963 commit dae9395

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

examples/models/mobilenet_v2/model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,8 @@
1515

1616

1717
class MV2Model(EagerModelBase):
18-
def __init__(self, useRealInput=True):
19-
self.useRealInput = useRealInput
18+
def __init__(self, use_real_input=True):
19+
self.use_real_input = use_real_input
2020
pass
2121

2222
def get_eager_model(self) -> torch.nn.Module:
@@ -28,7 +28,7 @@ def get_eager_model(self) -> torch.nn.Module:
2828
def get_example_inputs(self):
2929
tensor_size = (1, 3, 224, 224)
3030
input_batch = (torch.randn(tensor_size),)
31-
if self.useRealInput:
31+
if self.use_real_input:
3232
logging.info("Loaded real input image dog.jpg")
3333
import urllib
3434
url, filename = ("https://github.com/pytorch/hub/raw/master/images/dog.jpg", "dog.jpg")

0 commit comments

Comments
 (0)