- 在byol的config 中设置了 num_classes=1000: https://github.com/PaddlePaddle/PASSL/blob/9d7a9fd4af41772e29120553dddab1c162e4cb70/configs/byol/byol_r50_IM.yaml#L34 - 在model中设置了self.classifier = nn.Linear(embedding_dim, num_classes),并且forward中将classif_out和label一起传给了head  https://github.com/PaddlePaddle/PASSL/blob/9d7a9fd4af41772e29120553dddab1c162e4cb70/passl/modeling/architectures/BYOL.py#L263 - 在L2 Head中将对比loss和有监督的CE loss加在了一起返回  https://github.com/PaddlePaddle/PASSL/blob/9d7a9fd4af41772e29120553dddab1c162e4cb70/passl/modeling/heads/l2_head.py#L43
PASSL/configs/byol/byol_r50_IM.yaml
Line 34 in 9d7a9fd
PASSL/passl/modeling/architectures/BYOL.py
Line 263 in 9d7a9fd
PASSL/passl/modeling/heads/l2_head.py
Line 43 in 9d7a9fd