Very minor bug in TAKD
ashwinvaswani opened this issue · 1 comments
ashwinvaswani commented
In the train_distil_model function, the batch loss is logged instead of epoch loss.
Het-Shah commented
Yea this is a small error. @khizirsiddiqui can you rectify this?