TomTomTommi / HiNet

Official PyTorch implementation of "HiNet: Deep Image Hiding by Invertible Network" (ICCV 2021)
166 stars 35 forks source link

The loss value soar very high suddenly #12

Closed ByGary closed 2 years ago

ByGary commented 2 years ago

Hello! Recently I make an attempt to reproduce your work using the default hyperparameters (lamda_reconstruction = 5 lamda_guide = 1,lamda_low_frequency = 1, etc.) . But I observe that the loss value, especially the reconstruction loss will soar to extremely high, for example, from 0.0007 to 971213285 just between 10 mini-batches. Also, the revealed images turn to grean ones and appear terrible. Althouth the loss value may gradually decrease again, but that is vary strange and I can't figure it out. Did you encounter similar problems during your experiments? Could you explain a little bit and offer a possible solution? I will appreciate it if you reply.