tn_hidden_dim = 512
tn_dropout = 0.1
tn_nheads = 8
tn_dim_feedforward = 512
tn_enc_layers = 3
tn_dec_layers = 3