code: def check_torch_fsdp_plugin():
      if IS_FAST_TEST:
          registry = model_zoo.get_sub_registry(COMMON_MODELS)
      else:
          registry = model_zoo.get_sub_registry("transformers_gptj")
  
      for name, (
          model_fn,
          data_gen_fn,
          output_transform_fn,
          _,
          _,
      ) in registry.items():
          if any(
              element in name
              for element in [
                  "diffusers",
                  "deepfm_sparsearch",
                  "dlrm_interactionarch",
                  "torchvision_googlenet",
                  "torchvision_inception_v3",
              ]
          ):
              continue
          print(name)
          run_fn(model_fn, data_gen_fn, output_transform_fn)
          torch.cuda.empty_cache()
  
file_path: ColossalAI/tests/test_booster/test_plugin/test_torch_fsdp_plugin.py
imports: import pytest
  import torch
  from packaging import version
  from torch.optim import SGD
  import colossalai
  from colossalai.booster import Booster
  from colossalai.interface import OptimizerWrapper
  from colossalai.testing import clear_cache_before_run, rerun_if_address_is_in_use, spawn
  from tests.kit.model_zoo import COMMON_MODELS, IS_FAST_TEST, model_zoo
label_desc: Use image node_id 2 for calling a global function with example usage: check_torch_fsdp_plugin() without return types