关于aux_head.forward_train

[Wang-zhenyan]

作者您好： 非常感谢你的优秀的成果。我在学习您的代码过程中，想研究如何使用你所提出的contrast loss，但是我找不到代码中aux_head.forward_train（）在哪，您能够解答一下吗？