Implementation of correct number of samples

xuguodong03 / SSKD

[ECCV2020] Knowledge Distillation Meets Self-Supervision

233 stars 47 forks source link

Implementation of correct number of samples #1

Closed winycg closed 4 years ago

winycg commented 4 years ago

Hi, I look your implementation of counting correct number of augmented samples in student.py

254-tmp = torch.nonzero(rank, as_tuple=True)[0] 255-correct_num = tmp.numel()

Why I think correct_num represents the incorrect samples? Thanks!

xuguodong03 commented 4 years ago

Thank you for pointing the bugs! tmp.numel() does represent the incorrect samples. This may be the typo when I re-organised the code. And I have fixed it.