返回的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功能中進行設置並執行計算圖。