WisonZ / GKD

Code for GKD
8 stars 0 forks source link

When will the code be completed and released? #2

Closed Geek-lixiang closed 1 year ago

WisonZ commented 1 year ago

I have uploaded the loss function gkd_adaptive.py, and you can replace the vanilla kd loss with GKD loss. You may refer to TFace Github Link for a quick attempt.