Hi, I’m training a 2 class segmentation net on fastai2. My main code is,
////////////////////////////
learn.lr_find()
lr = 1e-3
learn.fit_flat_cos(1, slice(lr))
lr=1e-4
lrs = slice(lr)
learn.unfreeze()
learn.fit_flat_cos(3, lrs)
//////////////////////
When it runs to the last code line, I encounter a confusing thing. The train_loss and valid_loss of the first epoch are 0.4458 and 0.6682, but they become nan during the second epoch and the third epoch. And there are no any other errors being throw during the training process.
My configuration is:
fasiai 2.2.2
Are there something I don’t understand?
Thanks.