lenscloth / RKD

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

can't run the "run_distill.py" #4

Closed oxygenman closed 5 years ago

oxygenman commented 5 years ago

when I run the program follow the Quick Start,it failed with err as blow: Screenshot from 2019-07-29 18-15-11

I can't find what's wrong with it, could you help me ?

lenscloth commented 5 years ago

Thank you for showing interest in RKD,

For now, the source code does not support KD from GoogleNet => GoogleNet. You might try ResNet=>ResNet or ResNet => GoogleNet.

oxygenman commented 5 years ago

Thank you for showing interest in RKD,

For now, the source code does not support KD from GoogleNet => GoogleNet. You might try ResNet=>ResNet or ResNet => GoogleNet.

Okay, I have found it, you may better modify the Quick Start instruction, and by the way, thank you very much for your awesome work!

lenscloth commented 5 years ago

I changed the README. Thank you for reporting the issue :)