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