i was using trunc_normal_ for weight init... 😔