Hi,
First I really appreciated for your nice works.
I want to ask the inquiry about the sampling code.

In the RDMs paper, line 9 in pseudo code of 'Sampling from RDMs' says that Draw xe0,n ∼ Categorical(f(xt,n;θ)/τ);
However, in the code, I guess the performance of model can be reproduced when adding --argmax-decoding and it is different from the description.
Is it true that you turn on argmax-decoding mode when you do sampling?