def METHOD_NAME(self):
    # Some of our task models don't use the Backbone directly, but create
    # a feature extractor from it. In these cases, we don't want to count
    # the `backbone` as a layer, because it will be included in the model
    # summary and saves weights despite not being part of the model graph.
    METHOD_NAME = super().METHOD_NAME
    if hasattr(self, "backbone") and self.backbone in METHOD_NAME:
        # We know that the backbone is not part of the graph if it has no
        # inbound nodes.
        if len(self.backbone._inbound_nodes) == 0:
            METHOD_NAME.remove(self.backbone)
    return METHOD_NAME