search

NICALab / SUPPORT

Accurate denoising of voltage imaging data through statistically unbiased prediction, Nature Methods.
https://www.nature.com/articles/s41592-023-02005-8
GNU General Public License v3.0
68 stars 13 forks source link

Incorrect loss labels #11

Closed warnerwarner closed 11 months ago

warnerwarner commented 11 months ago

Hi,

I think that the losses are being incorrectly recorded - in the train function in src.train the l1, l2 and mean errors are incorrectly logged writer.add_scalar("Loss_l1/train_batch", loss_mean, epoch*len(train_dataloader) + i) writer.add_scalar("Loss_l2/train_batch", loss_mean_l1, epoch*len(train_dataloader) + i) writer.add_scalar("Loss/train_batch", loss_mean_l2, epoch*len(train_dataloader) + i)

EOMMINHO commented 11 months ago

Hi,

Thank you for letting us know the error! I have corrected it.