Implementation of Diversity Loss

[rayush7]

Hi @ucasligang

Thank you for your paper and code for SemMAE. Could you please point out the implementation of the diversity loss over attention maps (Equation 6 in the paper) in your code.