1
0
Fork 0

fix(api): correct name for kwargs

This commit is contained in:
Sean Sube 2023-01-22 17:25:29 -06:00
parent 986112efca
commit 9bff64c7c9
1 changed files with 2 additions and 2 deletions

View File

@ -75,10 +75,10 @@ class ONNXNet():
def half(self):
return self
def load_state_dict(self, _net, _strict=True) -> None:
def load_state_dict(self, net, strict=True) -> None:
pass
def to(self, _device):
def to(self, device):
return self