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:
|
||||
sess.use_deterministic_compute = True
|
||||
|
||||
return sess
|
||||
|
||||
def torch_str(self) -> str:
|
||||
if self.device.startswith("cuda"):
|
||||
return self.device
|
||||
|
|
Loading…
Reference in New Issue