Hi, thanks for the code release. I want to know which part is the teacher model in class FD, is the encoder or feature_model?