There are two required parameters for the nn.transformer module
(src - the sequence to the encoder (required) and tgt - the sequence to the decoder (required))
Is src (tgt) the length of a sentence or is it the length of a dictionary?
I already tried different options, but transformer absolutely does not want to learn.