Question about the query ranking.

[pkqbajng]

Hi, sir, thanks for your excellent job. Present implementation achieve the Adaptive Query Generation using torch.argsort and torch.gather. However, this will leading to that the parameters of the query ranking module won't update during training. You mentioned that this can be solved using gumble softmax, could you please share the code about this part?