为什么网络训练不起来

luting-hnu / EGRCNN

Edge-guided Recurrent Convolutional Neural Network for Multi-temporal Remote Sensing Image Building Change Detection

28 stars 0 forks source link

为什么网络训练不起来 #8

Open DoYoueHear opened 1 year ago

DoYoueHear commented 1 year ago

我用LEVIR-CD把每张10241024的原图裁剪成16张256256的小图，训练4、5个epoch，F1就发生下溢出，变成NAN。但是我通过transforms.resize把原图直接从10241024转换为256256，模型F1可以训练至75。裁剪成小图的训练集用A2Net可以正常训练、复现原论文结果。请问这是什么问题呢