Skip to content

Problem summing losses for VAE #983

@callum-b

Description

@callum-b

The VAE class in the example for VAE shows total_loss = reconstruction_loss + kl_loss, but this is not the case when adding these values in the fit output. I ran the source code from the example and ran it adapted to my own data, same mistake. How come? Should the loss calculations be adjusted during training?
vae_losses

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions