Open
Description
- 在byol的config 中设置了 num_classes=1000:
PASSL/configs/byol/byol_r50_IM.yaml
Line 34 in 9d7a9fd
- 在model中设置了self.classifier = nn.Linear(embedding_dim, num_classes),并且forward中将classif_out和label一起传给了head
PASSL/passl/modeling/architectures/BYOL.py
Line 263 in 9d7a9fd
- 在L2 Head中将对比loss和有监督的CE loss加在了一起返回
PASSL/passl/modeling/heads/l2_head.py
Line 43 in 9d7a9fd
Metadata
Assignees
Labels
No labels
Activity