Code Bug

[MrSchnappi]

annotated-transformer/the_annotated_transformer.py

Line 868 in debc9fd

memory, src_mask, ys, subsequent_mask(ys.size(1)).type_as(src.data)

Is there a problem with the parameters passed in here? It should be
out = test_model.decode(ys,memory, src_mask, subsequent_mask(ys.size(1)).type_as(src.data)) ,
not
out = test_model.decode(memory, src_mask, ys, subsequent_mask(ys.size(1)).type_as(src.data))