Forgot to transpose the nn layers.