search

lenscloth / RKD

Official pytorch Implementation of Relational Knowledge Distillation, CVPR 2019
395 stars 49 forks source link

torch type may not matching in pytorch==1.2 #7

Closed UnrealPawn closed 5 years ago

UnrealPawn commented 5 years ago

Hi~ I think there is a small problem in pairsampler.py. In pytorch ==1.0 works fine, but it's not work in pytorch == 1.2.

1569067728(1) bool type can not opt with uint8 type In pytorch == 1.2 you can change like this to fix it . 1569067931(1) ^——^

lenscloth commented 5 years ago

@UnrealPawn Thanks for noticing me the updates on pytorch 1.2.0! I will update if it becomes a fatal issue.