Skip to content

Commit 8fcd52e

Browse files
committed
mobilenet example
1 parent 31986ac commit 8fcd52e

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

examples/portable/module/module.cpp

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,14 +6,13 @@ using namespace ::executorch::extension;
66

77
int main(int argc, char** argv) {
88
// Create a Module.
9-
Module module(
10-
"/data/users/lfq/executorch/extension/module/test/resources/add.pte");
9+
Module module("/data/users/lfq/executorch/mobilenet_v3_small.pte");
1110

1211
// Wrap the input data with a Tensor.
13-
float input[1] = {4.f};
14-
auto tensor = from_blob(input, {1});
12+
float input[1 * 3 * 224 * 224] = {4.f};
13+
auto tensor = from_blob(input, {1, 3, 224, 224});
1514

16-
auto err = module.set_inputs({tensor, tensor});
15+
auto err = module.set_inputs({tensor});
1716

1817
// Perform an inference.
1918
const auto result = module.forward();

0 commit comments

Comments
 (0)