1
0
Fork 0

fix(api): add missing params to load stub

This commit is contained in:
Sean Sube 2023-01-16 10:59:47 -06:00
parent 9519fc16e9
commit fe657468bf
1 changed files with 1 additions and 1 deletions

View File

@ -48,7 +48,7 @@ class ONNXNet():
def half(self):
return self
def load_state_dict(self) -> None:
def load_state_dict(self, net, strict=True) -> None:
pass
def to(self, device):