21 lines
639 B
C++
21 lines
639 B
C++
#include <memory>
|
|
|
|
#include "AbstractInferencer.h"
|
|
#include "PianoInferencer.h"
|
|
#include "CPUInferencer.h"
|
|
|
|
using std::unique_ptr;
|
|
|
|
int main() {
|
|
using dynasty::inferencer::cpu::Inferencer;
|
|
auto inferencer =
|
|
Inferencer<float>::GetBuilder()->WithGraphOptimization(1)->WithONNXModel("res/example_prelu.origin.hdf5.onnx")->Build();
|
|
inferencer->Inference("res/input_config.json");
|
|
|
|
// This is to verify model with mystery node
|
|
// inferencer =
|
|
// Inferencer<float>::GetBuilder()->WithGraphOptimization(1)->WithONNXModel("res/mystery.onnx")->Build();
|
|
// inferencer->Inference("res/input_mystery.json");
|
|
|
|
}
|