return session options properly
This commit is contained in:
parent
5b4c370a1b
commit
881b290116
|
@ -117,6 +117,8 @@ class DeviceParams:
|
||||||
if "onnx-deterministic-compute" in self.optimizations:
|
if "onnx-deterministic-compute" in self.optimizations:
|
||||||
sess.use_deterministic_compute = True
|
sess.use_deterministic_compute = True
|
||||||
|
|
||||||
|
return sess
|
||||||
|
|
||||||
def torch_str(self) -> str:
|
def torch_str(self) -> str:
|
||||||
if self.device.startswith("cuda"):
|
if self.device.startswith("cuda"):
|
||||||
return self.device
|
return self.device
|
||||||
|
|
Loading…
Reference in New Issue