search

xiaomingzhid / sskd

Semi-supervised Knowledge Distillation
Apache License 2.0
43 stars 9 forks source link

Why the batch size must be larger than 56? #6

Closed ewigspace1910 closed 2 years ago

ewigspace1910 commented 2 years ago

@xiaomingzhid Line 85 in the file embedding_head.py:

if len(targets) > 56: cls_outputs = self.classifier(bn_feat, targets) else: cls_outputs = []

Could you explain why the targets (batch size?) must be larger than 56. Thanks