Skip to content

Commit b641827

Browse files
committed
some doc
1 parent f5677b4 commit b641827

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

users/zeyer/recog.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -168,8 +168,9 @@ def recog_model(
168168
:param recog_def:
169169
:param config:
170170
:param search_post_config:
171-
:param recog_post_proc_funcs: those are run before ``task.recog_post_proc_funcs``
172-
(which usually does BPE to words or so)
171+
:param recog_post_proc_funcs: Those are run before ``task.recog_post_proc_funcs``
172+
(which usually does BPE to words or so).
173+
Those are run after blank label removal and label repetition collapsing in case ``output_blank_label`` is set.
173174
:param search_mem_rqmt: for the search job. 6GB by default. can also be set via ``search_rqmt``
174175
:param search_rqmt: e.g. {"gpu": 1, "mem": 6, "cpu": 4, "gpu_mem": 24} or so
175176
:param dev_sets: which datasets to evaluate on. None means all defined by ``task``
@@ -235,7 +236,8 @@ def search_dataset(
235236
:param search_mem_rqmt: memory requirement for the search job
236237
:param search_rqmt: any additional requirements for the search job
237238
:param search_alias_name: alias name for the search job
238-
:param recog_post_proc_funcs: post processing functions for the recog output
239+
:param recog_post_proc_funcs: post processing functions for the recog output.
240+
Those are run after blank label removal and label repetition collapsing in case ``output_blank_label`` is set.
239241
:return: :class:`RecogOutput`, single best hyp (if there was a beam, we already took the best one)
240242
over the dataset
241243
"""

0 commit comments

Comments
 (0)