Question: Which loss metrics do they try in their new training procedure evaluated on the output of beam search?
Answer:  continuous relaxation to top-k-argmax