Python get_bert_optimizer 예제들

예제 #1

파일: biencoder.py 프로젝트: helderarr/WS

 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"),
     )

예제 #2

파일: train_cross_cands.py 프로젝트: rangell/BLINK

def get_optimizer(model, params):
    return get_bert_optimizer(
        [model],
        params["type_optimization"],
        params["learning_rate"],
        fp16=params.get("fp16"),
    )

예제 #3

 def get_optimizer(self):
     return get_bert_optimizer(
         [self.model],
         self.params["type_optimization"],
         self.params["learning_rate"],
         fp16=self.params.get("fp16"),
     )