def load(self, train_info: TrainingInfo) -> (dict, dict):
last_epoch = train_info.start_epoch_idx<EOL>model_state = torch.load(self.checkpoint_filename(last_epoch))<EOL>hidden_state = torch.load(self.checkpoint_hidden_filename(last_epoch))<EOL>self.checkpoint_strategy.restore(hidden_state)<EOL>train_info.restore(hidden_state)<EOL>return model_state, hidden_state<EOL>
Resume learning process and return loaded hidden state dictionary