Self_Explaining_Structures_Improve_NLP_Models icon indicating copy to clipboard operation
Self_Explaining_Structures_Improve_NLP_Models copied to clipboard

Published 20 hours ago • verified publisher icon ShannonAI

Cross entropy loss - reg loss?

Open Punchwes opened this issue 3 years ago • 0 comments

https://github.com/ShannonAI/Self_Explaining_Structures_Improve_NLP_Models/blob/d8b0511cd06964dbb23f1c4560f01996662b0b66/explain/trainer.py#L90

Hi, thanks for sharing your work. I am a little bit confused about how you calculate your final loss. In your paper, it says it should the sum but in the code it seems to be subtraction?

Punchwes avatar May 12 '21 09:05 Punchwes