docstr: Get pipeline layers for current stage.
docstr_tok: ["Get","pipeline","layers","for","current","stage","."]
code: def get_held_layers(self) -> List[Module]:
      """Get pipeline layers for current stage."""
      stage_manager = self.pipeline_stage_manager
      held_layers = super().get_held_layers()
      if stage_manager.is_last_stage():
          held_layers.append(self.model.score)
      return held_layers
  
imports: import warnings
  from functools import partial
  from typing import Callable, Dict, List
  from torch import Tensor, nn
  from torch.nn import Module
  import colossalai.shardformer.layer
  from ..modeling.falcon import FalconPipelineForwards, build_falcon_alibi_tensor_fn, get_falcon_flash_attention_forward, get_tp_falcon_decoder_layer_forward
  from .base_policy import ModulePolicyDescription, Policy, SubModuleReplacementDescription
label_desc: Use image node_id 3 for calling the FalconForSequenceClassificationPolicy obj's underlying member method code with example usage: obj.get_held_layers() and returns: held_layers