Hence, the continuous relaxation to top-k-argmax operation can be simply implemented by iteratively using the max operation which is continuous and allows for gradient flow during backpropagation. 
 Question: Which loss metrics do they try in their new training procedure evaluated on the output of beam search?
 continuous relaxation to top-k-argmax