15 lines
452 B
C++
15 lines
452 B
C++
#include <memory>
|
|
|
|
#include "AbstractInferencer.h"
|
|
#include "PianoInferencer.h"
|
|
#include "msft-gpu/MSFTInferencer.h"
|
|
|
|
using std::unique_ptr;
|
|
|
|
int main() {
|
|
using dynasty::inferencer::msftgpu::Inferencer;
|
|
auto inferencer =
|
|
Inferencer<float>::GetBuilder()->WithDeviceID(0) ->WithParallelLevel(1)->WithGraphOptimization(1)->WithONNXModel("res/example_prelu.origin.hdf5.onnx")->Build();
|
|
inferencer->Inference("res/input_config.json");
|
|
}
|