重新训练中途出现nan

nie-lang / UnsupervisedDeepImageStitching

TIP2021 - Unsupervised deep image stitching network

334 stars 50 forks source link

Open 12BEN34 opened 5 months ago

12BEN34 commented 5 months ago

对于融合网络，我使用自己的数据在提供的预训练模型上进行finetuning时，继续训练得到的模型不能重新融合，得到的图片全黑。另外自己重新训练时，也是同样的的问题，并且中途损失函数会出现nan