I am experimenting with Knowledge Distillation and wrote my Distiller class following the reference -https://keras.io/examples/vision/knowledge_distillation/
When I t