Can your implementation run and how to run?

[LiangHann]

Can your implementation run and how to run?

[UdbhavPrasad072300]

There are notebook examples for each transformer in the notebooks directory. The parameters of the model are quite straight forward: s_vocab_size, t_vocab_size, embed_size, num_heads, num_ff, encode_layers, decode_layers, hidden_size, dropout=0.2, device="cpu")