Solved by b02202050 in post #2 
              
              
                You can override the function without actually modifying the source code like the following. 
import efficientnet_pytorch
def my_efficientnet_forward(self, inputs):
	""" Calls extract_features to extract features, applies final linear layer, and returns logits. """
	bs = inputs.size(0)
	# Convolutio…