About training loss, Where is the Gradient Penalty? #478

Open
opened 2024-12-14 05:47:31 +01:00 by QinjieXiao · 0 comments
QinjieXiao commented 2024-12-14 05:47:31 +01:00 (Migrated from github.com)

Thanks for releasing the excellent work. I have some questions about the training loss.
Where is the Gradient Penalty? I found some logs about "Gradient Penalty", e.g., D_GP in the released ckpt, but I cannot find code in the projected_model.py.
Also, a found a gp implementation in fs_model.py, but it was not use for training.

Thanks for releasing the excellent work. I have some questions about the training loss. Where is the Gradient Penalty? I found some logs about "Gradient Penalty", e.g., D_GP in the released ckpt, but I cannot find code in the projected_model.py. Also, a found a gp implementation in fs_model.py, but it was not use for training.
Sign in to join this conversation.