llmcompressor.modifiers.pruning.utils.pytorch

模块

• layer_mask –

函数

• param_mask_name –

  用于稀疏层的掩码缓冲区的名称

param_mask_name

param_mask_name() -> str

用于稀疏层的掩码缓冲区的名称

源代码在 llmcompressor/modifiers/pruning/utils/pytorch/layer_mask.py

def param_mask_name() -> str:
    """
    Name to use for mask buffer on a sparse layer
    """
    return "mask"