You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
While trying to repro the in-painting notebook (really cool by the way), I noticed u-net is loading a frozen pre-trained imagenet stem: learn = unet_learner(dbunch, partial(m, sa=sa), opt_func=opt_func, metrics=[], loss_func=MSELoss())
While trying to repro the in-painting notebook (really cool by the way), I noticed u-net is loading a frozen pre-trained imagenet stem:
learn = unet_learner(dbunch, partial(m, sa=sa), opt_func=opt_func, metrics=[], loss_func=MSELoss())
In this case, the default
pretrained
flag needs to be specifically turned off:https://github.com/fastai/fastai2/blob/master/fastai2/vision/learner.py#L183
From my initial tests, it seems to matter.
@JoshVarty
The text was updated successfully, but these errors were encountered: