MCPcopy
hub / github.com/thunlp/OpenKE / save_checkpoint

Method save_checkpoint

openke/module/BaseModule.py:20–21  ·  view source on GitHub ↗
(self, path)

Source from the content-addressed store, hash-verified

18 self.eval()
19
20 def save_checkpoint(self, path):
21 torch.save(self.state_dict(), path)
22
23 def load_parameters(self, path):
24 f = open(path, "r")

Calls

no outgoing calls

Tested by

no test coverage detected