estava a usar trunc_normal_ para inicialização de pesos... 😔