def load(self):
"""Load the model from the last checkpoint."""
if not self.load_path.exists():
log.info("[initializing new `{}`]".format(self.__class__.__name__))
self.dict.init_from_scratch()
self.embdict.init_from_scratch(self.dict.tok2int_vocab)
if hasattr(self.dict, 'char2int_vocab'):
chars_num = len(self.dict.char2int_vocab)
else:
chars_num = 0
self._net = RankingNetwork(chars_num=chars_num,
toks_num=len(self.dict.tok2int_vocab),
emb_dict=self.embdict,
**self.network_parameters)
self._net.init_from_scratch(self.embdict.emb_matrix)
else:
log.info("[initializing `{}` from saved]".format(
self.__class__.__name__))
self.dict.load()
self.embdict.load()
if hasattr(self.dict, 'char2int_vocab'):
chars_num = len(self.dict.char2int_vocab)
else:
chars_num = 0
self._net = RankingNetwork(chars_num=chars_num,
toks_num=len(self.dict.tok2int_vocab),
emb_dict=self.embdict,
**self.network_parameters)
self._net.load(self.load_path)
def load(self):
if not self.load_path.exists():
log.info("[initializing new `{}`]".format(self.__class__.__name__))
self.dict.init_from_scratch()
self._net = RankingNetwork(len(self.dict.tok2int_vocab), **self.network_parameters)
embdict_parameter_names = list(inspect.signature(Embeddings.__init__).parameters)
embdict_parameters = {par: self.opt[par] for par in embdict_parameter_names if par in self.opt}
embdict= Embeddings(self.dict.tok2int_vocab, **embdict_parameters)
self._net.set_emb_matrix(embdict.emb_matrix)
else:
log.info("[initializing `{}` from saved]".format(self.__class__.__name__))
self.dict.load()
self._net = RankingNetwork(len(self.dict.tok2int_vocab), **self.network_parameters)
self._net.load(self.load_path)
def load(self):
if not self.load_path.exists():
log.info("[initializing new `{}`]".format(self.__class__.__name__))
self.dict.init_from_scratch()
self.embdict.init_from_scratch(self.dict.tok2int_vocab)
self._net = RankingNetwork(toks_num=len(self.dict.tok2int_vocab),
emb_dict=self.embdict,
**self.network_parameters)
self._net.init_from_scratch(self.embdict.emb_matrix)
else:
log.info("[initializing `{}` from saved]".format(self.__class__.__name__))
self.dict.load()
self.embdict.load()
self._net = RankingNetwork(toks_num=len(self.dict.tok2int_vocab),
emb_dict=self.embdict,
**self.network_parameters)
self._net.load(self.load_path)