Compute G_Loss called twice

cyclomon / UNSB

Official Repository of "Unpaired Image-to-Image Translation via Neural Schrödinger Bridge" (ICLR 2024)

MIT License

154 stars 5 forks source link

Closed juanprietob closed 7 months ago

juanprietob commented 7 months ago

I noticed that the compute_G_loss() function is called twice during the optimization. https://github.com/cyclomon/UNSB/blob/main/models/sb_model.py#L143-L144

cyclomon commented 7 months ago

Thank you for your comment! We corrected the redundant part.