def load_state_dict(self, state_dict):
    state_dict = deepcopy(state_dict)
    self.wait_count = state_dict['wait_count']
    self.stopped_epoch = state_dict['stopped_epoch']
    self.best_score = state_dict['best_score']
    self.patience = state_dict['patience']
def load_state_dict(self, state_dict):
    state_dict = deepcopy(state_dict)
    self.wait_count = state_dict['wait_count']
    self.stopped_epoch = state_dict['stopped_epoch']
    self.best_score = state_dict['best_score']
    self.patience = state_dict['patience']