My multi-dimensional Transformer doesn't seem to learn anything

[utkuumetin (Utku Metin)]

Hi, i’m facing similar issue. Mine gives me equal probability for every output.