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)