Skip to content

BYOL的预训练中好像使用了gt_label? #134

Open
@youqingxiaozhua

Description

  • 在byol的config 中设置了 num_classes=1000:
    num_classes: 1000
  • 在model中设置了self.classifier = nn.Linear(embedding_dim, num_classes),并且forward中将classif_out和label一起传给了head

image

outputs = self.head(a1, b1, a2, b2, classif_out, label)

  • 在L2 Head中将对比loss和有监督的CE loss加在了一起返回

image

outputs["loss"] = outputs['rep_loss'] + outputs["class_loss"]

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions