fix convert to onnx node's execution options

chaiNNer-org · Sep 5, 2022 · cf80665 · cf80665
1 parent 75e1c04
commit cf80665
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend/src/nodes/pytorch_nodes.py b/backend/src/nodes/pytorch_nodes.py
@@ -477,7 +477,7 @@ def __init__(self):
             pass
 
     def run(self, model: torch.nn.Module) -> bytes:
-        exec_options = get_execution_options()
+        exec_options = to_pytorch_execution_options(get_execution_options())
 
         model = model.eval()
         if exec_options.device == "cuda":