File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -168,8 +168,9 @@ def recog_model(
168
168
:param recog_def:
169
169
:param config:
170
170
: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.
173
174
:param search_mem_rqmt: for the search job. 6GB by default. can also be set via ``search_rqmt``
174
175
:param search_rqmt: e.g. {"gpu": 1, "mem": 6, "cpu": 4, "gpu_mem": 24} or so
175
176
:param dev_sets: which datasets to evaluate on. None means all defined by ``task``
@@ -235,7 +236,8 @@ def search_dataset(
235
236
:param search_mem_rqmt: memory requirement for the search job
236
237
:param search_rqmt: any additional requirements for the search job
237
238
: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.
239
241
:return: :class:`RecogOutput`, single best hyp (if there was a beam, we already took the best one)
240
242
over the dataset
241
243
"""
You can’t perform that action at this time.
0 commit comments