Commit a445c718 authored by Mao, Bojia's avatar Mao, Bojia
Browse files

recorrect onnx to device

parent d735c192
......@@ -92,7 +92,7 @@ class ONNXInterface:
ort_outs = self.ort_session.run([output_name], ort_inputs)
return ort_outs
def to(self,decive):
def to(self,device):
self.device = device
if self.defaultmodel != None:
self.defaultmodel.to(device)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment