From eccf7cbaf9bf267b2ae2d24768957b6abd3039ce Mon Sep 17 00:00:00 2001 From: BujSet Date: Thu, 24 Jul 2025 19:46:05 +0000 Subject: [PATCH 1/5] Moving some of the setup logic straight into the docker build --- .ci/docker/common/install_zephyr.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/.ci/docker/common/install_zephyr.sh b/.ci/docker/common/install_zephyr.sh index c24bb5aa8f1..86c80960805 100644 --- a/.ci/docker/common/install_zephyr.sh +++ b/.ci/docker/common/install_zephyr.sh @@ -84,4 +84,5 @@ install_prerequiresites() { pip_install pyelftools } + install_prerequiresites From 885f1d93b90a3b4423f0d94bcf057eed76a7c7ad Mon Sep 17 00:00:00 2001 From: BujSet Date: Wed, 30 Jul 2025 15:33:36 -0700 Subject: [PATCH 2/5] using dog.jpg as real input default --- examples/models/mobilenet_v2/model.py | 24 +++++++++++++++++++++++- 1 file changed, 23 insertions(+), 1 deletion(-) diff --git a/examples/models/mobilenet_v2/model.py b/examples/models/mobilenet_v2/model.py index f15178ac71b..ec6e3abb0dd 100644 --- a/examples/models/mobilenet_v2/model.py +++ b/examples/models/mobilenet_v2/model.py @@ -15,7 +15,8 @@ class MV2Model(EagerModelBase): - def __init__(self): + def __init__(self, useRealInput=True): + self.useRealInput = useRealInput pass def get_eager_model(self) -> torch.nn.Module: @@ -26,6 +27,27 @@ def get_eager_model(self) -> torch.nn.Module: def get_example_inputs(self): tensor_size = (1, 3, 224, 224) + input_batch = (torch.randn(tensor_size),) + if self.useRealInput: + logging.info("Loaded real input image dog.jpg") + import urllib + url, filename = ("https://github.com/pytorch/hub/raw/master/images/dog.jpg", "dog.jpg") + try: + urllib.URLopener().retrieve(url, filename) + except: + urllib.request.urlretrieve(url, filename) + from PIL import Image + from torchvision import transforms + input_image = Image.open(filename) + preprocess = transforms.Compose([ + transforms.Resize(256), + transforms.CenterCrop(224), + transforms.ToTensor(), + transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]), + ]) + input_tensor = preprocess(input_image) + input_batch = input_tensor.unsqueeze(0) + input_batch = (input_batch,) return (torch.randn(tensor_size),) From e0e98a3467e46e2831767eb8e6cbfc75deeb9890 Mon Sep 17 00:00:00 2001 From: BujSet Date: Wed, 30 Jul 2025 15:40:38 -0700 Subject: [PATCH 3/5] Removing unrelated change --- .ci/docker/common/install_zephyr.sh | 1 - 1 file changed, 1 deletion(-) diff --git a/.ci/docker/common/install_zephyr.sh b/.ci/docker/common/install_zephyr.sh index 86c80960805..c24bb5aa8f1 100644 --- a/.ci/docker/common/install_zephyr.sh +++ b/.ci/docker/common/install_zephyr.sh @@ -84,5 +84,4 @@ install_prerequiresites() { pip_install pyelftools } - install_prerequiresites From 5afc33611eee440d2062f04e79f3113adc286c62 Mon Sep 17 00:00:00 2001 From: BujSet Date: Wed, 30 Jul 2025 16:00:06 -0700 Subject: [PATCH 4/5] Naming convention fix --- examples/models/mobilenet_v2/model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/models/mobilenet_v2/model.py b/examples/models/mobilenet_v2/model.py index ec6e3abb0dd..e2fbb4ad658 100644 --- a/examples/models/mobilenet_v2/model.py +++ b/examples/models/mobilenet_v2/model.py @@ -15,8 +15,8 @@ class MV2Model(EagerModelBase): - def __init__(self, useRealInput=True): - self.useRealInput = useRealInput + def __init__(self, use_real_input=True): + self.use_real_input = use_real_input pass def get_eager_model(self) -> torch.nn.Module: @@ -28,7 +28,7 @@ def get_eager_model(self) -> torch.nn.Module: def get_example_inputs(self): tensor_size = (1, 3, 224, 224) input_batch = (torch.randn(tensor_size),) - if self.useRealInput: + if self.use_real_input: logging.info("Loaded real input image dog.jpg") import urllib url, filename = ("https://github.com/pytorch/hub/raw/master/images/dog.jpg", "dog.jpg") From 7883a76836e7518c4828043e2862310b9f827b36 Mon Sep 17 00:00:00 2001 From: BujSet Date: Thu, 7 Aug 2025 17:50:09 +0000 Subject: [PATCH 5/5] Linting --- examples/models/mobilenet_v2/model.py | 27 ++++++++++++++++++--------- 1 file changed, 18 insertions(+), 9 deletions(-) diff --git a/examples/models/mobilenet_v2/model.py b/examples/models/mobilenet_v2/model.py index e2fbb4ad658..5c2c7ff7016 100644 --- a/examples/models/mobilenet_v2/model.py +++ b/examples/models/mobilenet_v2/model.py @@ -31,20 +31,29 @@ def get_example_inputs(self): if self.use_real_input: logging.info("Loaded real input image dog.jpg") import urllib - url, filename = ("https://github.com/pytorch/hub/raw/master/images/dog.jpg", "dog.jpg") - try: + + url, filename = ( + "https://github.com/pytorch/hub/raw/master/images/dog.jpg", + "dog.jpg", + ) + try: urllib.URLopener().retrieve(url, filename) - except: + except: urllib.request.urlretrieve(url, filename) from PIL import Image from torchvision import transforms + input_image = Image.open(filename) - preprocess = transforms.Compose([ - transforms.Resize(256), - transforms.CenterCrop(224), - transforms.ToTensor(), - transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]), - ]) + preprocess = transforms.Compose( + [ + transforms.Resize(256), + transforms.CenterCrop(224), + transforms.ToTensor(), + transforms.Normalize( + mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225] + ), + ] + ) input_tensor = preprocess(input_image) input_batch = input_tensor.unsqueeze(0) input_batch = (input_batch,)