Skip to content

Commit 50e48f1

Browse files
dboydboyliao
authored andcommitted
Add model file
1 parent c74d264 commit 50e48f1

File tree

4 files changed

+258
-168
lines changed

4 files changed

+258
-168
lines changed

main.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ int argmax(const Tensor &logits) {
2323
return max_index;
2424
}
2525

26-
static My_model model;
26+
static MyModel model;
2727

2828
int main(int argc, char **argv) {
2929
printf("\n");
@@ -35,8 +35,8 @@ int main(int argc, char **argv) {
3535
Tensor input_image = new RomTensor({1, 28, 28, 1}, flt, arr_input_image[i]);
3636
Tensor logits = new RamTensor({1, 10}, flt);
3737

38-
model.set_inputs({{My_model::input_0, input_image}})
39-
.set_outputs({{My_model::output_0, logits}})
38+
model.set_inputs({{MyModel::input_0, input_image}})
39+
.set_outputs({{MyModel::output_0, logits}})
4040
.eval();
4141
int max_index = argmax(logits);
4242
input_image.free();

0 commit comments

Comments
 (0)