[Flux] Normalize validation loss by latent elements (#4195)