Hello, firstly thank you for sharing such an excellent work.
When I was running the code you provided, I found that there is a bit of a problem in the part of the code for the calculation of clf_loss. The code uses the same loss function as in the paper - cross entropy loss. In this case, for the calculation of clf_loss in the screenshot below, should the soft_label be replaced with closest_mode_indices. squeeze(), or do you have another calculation method, hope to get your reply, thanks again for sharing your excellent work.

Hello, firstly thank you for sharing such an excellent work.

When I was running the code you provided, I found that there is a bit of a problem in the part of the code for the calculation of clf_loss. The code uses the same loss function as in the paper - cross entropy loss. In this case, for the calculation of clf_loss in the screenshot below, should the soft_label be replaced with closest_mode_indices. squeeze(), or do you have another calculation method, hope to get your reply, thanks again for sharing your excellent work.