I have made a switch from TensorFlow to PyTorch recently. I use a famous Github repo for my EfficientNet. However, when I wrote the model initiation class as follows: