Skip to content

Files

Latest commit

Apr 18, 2025
4f58fc9 · Apr 18, 2025

History

History
66 lines (35 loc) · 1.88 KB

modeling_utils.md

File metadata and controls

66 lines (35 loc) · 1.88 KB

自定义层和工具

此页面列出了库使用的所有自定义层,以及它为模型提供的实用函数。

其中大多数只有在您研究库中模型的代码时才有用。

Pytorch自定义模块

[[autodoc]] pytorch_utils.Conv1D

PyTorch帮助函数

[[autodoc]] pytorch_utils.apply_chunking_to_forward

[[autodoc]] pytorch_utils.find_pruneable_heads_and_indices

[[autodoc]] pytorch_utils.prune_layer

[[autodoc]] pytorch_utils.prune_conv1d_layer

[[autodoc]] pytorch_utils.prune_linear_layer

TensorFlow自定义层

[[autodoc]] modeling_tf_utils.TFConv1D

[[autodoc]] modeling_tf_utils.TFSequenceSummary

TensorFlow loss 函数

[[autodoc]] modeling_tf_utils.TFCausalLanguageModelingLoss

[[autodoc]] modeling_tf_utils.TFMaskedLanguageModelingLoss

[[autodoc]] modeling_tf_utils.TFMultipleChoiceLoss

[[autodoc]] modeling_tf_utils.TFQuestionAnsweringLoss

[[autodoc]] modeling_tf_utils.TFSequenceClassificationLoss

[[autodoc]] modeling_tf_utils.TFTokenClassificationLoss

TensorFlow帮助函数

[[autodoc]] modeling_tf_utils.get_initializer

[[autodoc]] modeling_tf_utils.keras_serializable

[[autodoc]] modeling_tf_utils.shape_list