Loss

[hongzimao]

Shouldn't it be reduced_mean instead of reduced_sum when computing the vae and reconstruction loss?

[atabakd]

Yes, but if you use reduced_sum for all losses consistently, then it shouldn't matter (mean is basically sum, divided by a constant)