How to use Pytorch to create a custom EfficientNet with the last layer written correctly

Question

I have a classification problem to predict 8 classes for example, I am using EfficientNetB3 in pytorch from here. However, I got confused on whether my custom c