Shengcao-Cao / CMT

[CVPR 2023] CMT: Contrastive Mean Teacher for Domain Adaptive Object Detectors
Apache License 2.0
37 stars 6 forks source link

Where is the definition of contrastive learning loss? I can't find it. #13

Closed suojinhui closed 4 months ago

suojinhui commented 4 months ago

Where is the definition of contrastive learning loss? I can't find it.

Shengcao-Cao commented 4 months ago

Hello @suojinhui , you may find the contrastive learning loss here: https://github.com/Shengcao-Cao/CMT/blob/029cd21d0d77252d1309f87b83c05d1752055e56/CMT_AT/adapteacher/engine/trainer.py#L784

suojinhui commented 4 months ago

Thank you for your reply!