code: def load_efficientnet(num_classes, model_name="efficientnet-b0"):
      """Loads the EfficientNet from TF Model Garden."""
      overrides = {
          "num_classes": num_classes,
      }
      return efficientnet_model.EfficientNet.from_name(
          model_name, overrides=overrides
      )
  
imports: import os
  import pickle
  from typing import Any, Dict, Optional
  import uuid
  from ..None import machine_config
  from ..None import run
  import tensorflow
  import tensorflow_datasets
  from official.legacy.image_classification.efficientnet import efficientnet_model
  from official.legacy.image_classification.resnet import resnet_model
label_desc: Use image node_id 8 for calling a global function with example usage: load_efficientnet(num_classes, model_name) and returns: efficientnet_model