SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

push (#9008)

Browse Source
This commit is contained in:
Patrick von Platen
2020-12-09 15:14:33 +01:00
committed by GitHub
parent 61abd50b98
commit da37a21c89
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
docs/source/internal/generation_utils.rst
View File

@@ -15,7 +15,8 @@ Utilities for Generation
This page lists all the utility functions used by :meth:`~transformers.PretrainedModel.generate`, This page lists all the utility functions used by :meth:`~transformers.PretrainedModel.generate`,
:meth:`~transformers.PretrainedModel.greedy_search`, :meth:`~transformers.PretrainedModel.sample`, :meth:`~transformers.PretrainedModel.greedy_search`, :meth:`~transformers.PretrainedModel.sample`,
:meth:`~transformers.PretrainedModel.beam_search`, and :meth:`~transformers.PretrainedModel.beam_sample`. :meth:`~transformers.PretrainedModel.beam_search`, :meth:`~transformers.PretrainedModel.beam_sample`, and
:meth:`~transformers.PretrainedModel.group_beam_search`.
Most of those are only useful if you are studying the code of the generate methods in the library. Most of those are only useful if you are studying the code of the generate methods in the library.

4
src/transformers/generation_logits_process.py
View File

@@ -415,8 +415,8 @@ class PrefixConstrainedLogitsProcessor(LogitsProcessor):
class HammingDiversityLogitsProcessor(LogitsProcessor): class HammingDiversityLogitsProcessor(LogitsProcessor):
r""" r"""
:class:`transformers.LogitsProcessor` that enforces diverse beam search. Note that this logits processor is only :class:`transformers.LogitsProcessor` that enforces diverse beam search. Note that this logits processor is only
effective for `group_beam_search`. See `Diverse Beam Search: Decoding Diverse Solutions from Neural Sequence Models effective for :meth:`transformers.PretrainedModel.group_beam_search`. See `Diverse Beam Search: Decoding Diverse
<https://arxiv.org/pdf/1610.02424.pdf>`__ for more details. Solutions from Neural Sequence Models <https://arxiv.org/pdf/1610.02424.pdf>`__ for more details.
Args: Args:
diversity_penalty (:obj:`float`): diversity_penalty (:obj:`float`):