Browse Source

feat(load_and_run): feat --input parameter

GitOrigin-RevId: 2df8b6c164
release-1.1
Megvii Engine Team 4 years ago
parent
commit
65c549137f
1 changed files with 9 additions and 6 deletions
  1. +9
    -6
      sdk/load-and-run/src/mgblar.cpp

+ 9
- 6
sdk/load-and-run/src/mgblar.cpp View File

@@ -807,13 +807,16 @@ void run_test_st(Args &env) {
parser.feed(path);
}
auto inputs = parser.inputs;
for (auto& i : inputs) {
if (tensormap.find(i.first) == tensormap.end()) {
continue;
}
if (inputs.size() > 1) {
for (auto& i : inputs) {
mgb_assert(tensormap.find(i.first) != tensormap.end());

auto& in = tensormap.find(i.first)->second;
in->copy_from(i.second);
auto& in = tensormap.find(i.first)->second;
in->copy_from(i.second);
}
} else {
auto& in = tensormap.begin()->second;
in->copy_from(inputs.begin()->second);
}

warmup();


Loading…
Cancel
Save