About training loss, Where is the Gradient Penalty?

[QinjieXiao]

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.