def __init__(self, hparams: DictConfig): super().__init__() self.hparams = hparams self.model = load_bert_fold(self.hp) if self.hp.use_ema: self.ema_model = create_ema(self.model)
def __init__(self, hparams: DictConfig): super().__init__() self.hparams = hparams self.model = MobileNetV2_unet( drop_rate=self.hp.drop_rate, drop_path_rate=self.hp.drop_path_rate, ) self.criterion = dice_loss(scale=2) if self.hp.use_ema: self.ema_model = create_ema(self.model)