def METHOD_NAME(self, optim_states=None, saved_optim_type=None):
    return get_bert_optimizer(
        [self.model],
        self.params["type_optimization"],
        self.params["learning_rate"],
        fp16=self.params.get("fp16"),
    )