Skip to content

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

@youqingxiaozhua

Description

@youqingxiaozhua
  • 在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"]

Metadata

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