返回的BeamSearch解码代码库

Question

此配置示例使用了从ctc或seq2seq模型向后添加lm_score的示例用法https://github.com/rwth-i6/returnn-experiments/blob/master/2018-asr-attention/librispeech/attention/exp3.ctc.lm的.config

我想知道在波束搜索解码中如何使用它。 我找不到BeamSearch解码示例吗？ 指向实现它的代码的指针将很有用。

Answer 1

该配置描述了模型，以及一些用于训练和/或解码的超级参数。

执行训练和/或解码的实际代码在Returnn本身中。 请参阅完整设置作为示例，说明如何调用Returnn进行光束搜索。

很简单，在Returnn中，波束搜索是通过纯TF函数实现的，因此它将在TF计算图中运行。 在为模型构建计算图时，有search_flag表示应该执行搜索。 如果设置了search_flag则ChoiceLayer将通过tf.topk扩展搜索范围。 在TFEngine的search功能中进行设置并执行计算图。