Rohan_Kumar:

,d , ‘eos’] this is trg inp, since it (seq len, bz ) dim, i do t


Sorry. I’m confused. In a WLM problem, the target input is [a , b, c] and the output is [d-ish], which is compared against [d]. [d-ish] is something from the model during the training.