First of all, thank you for the implementation. My issue is that I'm trying to modify the feature extraction part using EfficientNet. However, I'm encountering errors related to selecting specific layers on the teacher model. Could you help me with that? Thanks in advance!

[AbelAbeb]