diff --git a/src/transformers/models/efficientnet/modeling_efficientnet.py b/src/transformers/models/efficientnet/modeling_efficientnet.py index 5b7ff534eedfe4..e415d7f1b46a1e 100644 --- a/src/transformers/models/efficientnet/modeling_efficientnet.py +++ b/src/transformers/models/efficientnet/modeling_efficientnet.py @@ -484,6 +484,7 @@ class EfficientNetPreTrainedModel(PreTrainedModel): config_class = EfficientNetConfig base_model_prefix = "efficientnet" main_input_name = "pixel_values" + _no_split_modules = [] def _init_weights(self, module): """Initialize the weights"""