search

QQiuyp / FIN

This is the source code of paper FIN: Flow-based Robust Watermarking with Invertible Noise Layer for Black-box Distortions, which is received by AAAI' 23.
48 stars 5 forks source link

训练一段时间后出现了nan #8

Open zxx01 opened 10 months ago

zxx01 commented 10 months ago

屏幕截图 2024-01-07 214730 这是怎么回事？

QQiuyp commented 10 months ago

In our experiments, we also sometimes encounter such instability issues, and this unstable training process in INN-based models has been mentioned in other literature as well. In this paper, our solution is to adjust the weights of the message loss and PSNR loss once the accuracy exceeds 95% or nan occurs. Specifically, we reload the model parameters from before the crash and, in the new training session, decrease the weight of the message loss and increase the weight of the PSNR loss.

zxx01 commented 10 months ago

OK, Thanks!