Skip to content

Instantly share code, notes, and snippets.

@neelriyer

neelriyer/create_unet_learner.py

Last active July 5, 2020 10:06
Show Gist options
  • Save neelriyer/dbaf384c9f2e3af5a9b80133ee304c53 to your computer and use it in GitHub Desktop.
Save neelriyer/dbaf384c9f2e3af5a9b80133ee304c53 to your computer and use it in GitHub Desktop.
Download ZIP
Initial train for video restoration
Raw
create_unet_learner.py
data = get_data(bs=64,size=128)
learn = None
gc.collect()
wd = 1e-3
learn = unet_learner(data, arch, wd=wd, loss_func=feat_loss, callback_fns=LossMetrics,blur=True, norm_type=NormType.Weight)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment