def save(self, checkpoint_path, best_err, curr_err):
    state = {'lr': self.lr, 'best_err': best_err, 'curr_err': curr_err, 'model': self.model.state_dict(), 'optimizer': self.optimizer.state_dict()}
    torch.save(state, checkpoint_path)
def save(self, checkpoint_path, best_err, curr_err):
    state = {'lr': self.lr, 'best_err': best_err, 'curr_err': curr_err, 'model': self.model.state_dict(), 'optimizer': self.optimizer.state_dict()}
    torch.save(state, checkpoint_path)