code: def is_se_weight(key, value):
      return key.endswith("squeeze.weight") or key.endswith(
          "expand.weight"
      )
  
file_path: DALI/docs/examples/use_cases/pytorch/efficientnet/image_classification/models/model.py
imports: from dataclasses import dataclass, asdict, replace
  from .common import SequentialSqueezeAndExcitationTRT, SequentialSqueezeAndExcitation, SqueezeAndExcitation, SqueezeAndExcitationTRT
  from typing import Optional, Callable
  import os
  import torch
  import argparse
  from functools import partial
label_desc: Use image node_id 2 for calling a global function with example usage: is_se_weight(key, value) and returns: unknown