generated from fastai/nbdev_template
-
Notifications
You must be signed in to change notification settings - Fork 51
Open
Description
currently using unet_config . An obsolete method for learner params
#Problem:
cfg = unet_config(blur=True, norm_type=NormType.Weight, self_attention=True,
y_range=y_range)
def create_gen_learner():
return unet_learner(dls_gen, bbone, loss_func=loss_gen, config=cfg)
#Fix: remove cfg lne & Feed the params in directly to the learner
def create_gen_learner():
return unet_learner(dls_gen, bbone, loss_func=loss_gen,
blur=True, norm_type=NormType.Weight, self_attention=True,
y_range=y_range)
Metadata
Metadata
Assignees
Labels
No labels