def get_optimizer(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"), )
def get_optimizer(model, params): return get_bert_optimizer( [model], params["type_optimization"], params["learning_rate"], fp16=params.get("fp16"), )
def get_optimizer(self): return get_bert_optimizer( [self.model], self.params["type_optimization"], self.params["learning_rate"], fp16=self.params.get("fp16"), )