fix(api): add missing params to load stub
This commit is contained in:
parent
9519fc16e9
commit
fe657468bf
|
@ -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):
|
||||
|
|
Loading…
Reference in New Issue