Commit 4c00584
Adding Models to Test Matrix for Zephyr CI Test (pytorch#13193)
### Summary
Previously, only the example Add model was included in the matrix of
models that the Zephyr CI job tests. The Add model isn't very
representative of real-world models, and may give a false sense of
security on the functionality of incorporating ExecuTorch with Zephyr.
This PR adds the simple `Softmax` and `Mobilenet V2` models to the test
matrix. While the `Softmax` model is like the `Add` model (a toy
example), `Mobilenet V2` includes using an image from the `imagenet`
validation set, and verifies that the top predicted label matches
expectation (i.e. predict "Samoyed" as best label). All three of these
tests show example of this flow can be extend further for other models.
Thus this PR integrates these test to ensure compatibility between
ExecuTorch and Zephyr is maintained.
### Test plan
Verified that these models work with the Zephyr ecosystem manually in a
separate docker image.1 parent b89243e commit 4c00584
1 file changed
+14
-6
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
60 | 60 | | |
61 | 61 | | |
62 | 62 | | |
63 | | - | |
| 63 | + | |
64 | 64 | | |
65 | 65 | | |
66 | 66 | | |
| |||
72 | 72 | | |
73 | 73 | | |
74 | 74 | | |
| 75 | + | |
| 76 | + | |
| 77 | + | |
| 78 | + | |
| 79 | + | |
| 80 | + | |
| 81 | + | |
| 82 | + | |
| 83 | + | |
| 84 | + | |
75 | 85 | | |
76 | 86 | | |
77 | 87 | | |
| |||
118 | 128 | | |
119 | 129 | | |
120 | 130 | | |
121 | | - | |
| 131 | + | |
122 | 132 | | |
123 | 133 | | |
124 | 134 | | |
125 | 135 | | |
126 | | - | |
127 | 136 | | |
128 | | - | |
| 137 | + | |
129 | 138 | | |
130 | 139 | | |
131 | 140 | | |
132 | 141 | | |
133 | 142 | | |
134 | 143 | | |
135 | 144 | | |
136 | | - | |
137 | 145 | | |
138 | | - | |
| 146 | + | |
139 | 147 | | |
140 | 148 | | |
141 | 149 | | |
| |||
0 commit comments