Thanks for your code! From what I know, fitnet uses "hints" and both KD loss seems like KDCriterion only use <hint> module as criterion is there something I might be missing? thanks!