Skip to content

Add support for multiple losses during training #23

Description

@george-gca

Is your feature request related to a problem? Please describe.
My model uses multiple losses during training, and it would be useful if I could visualize each one of them.

Describe the solution you'd like
When sending info to server, it would be nice if a dict of losses (composed by its names and current value) could be provided.

Describe alternatives you've considered
Maybe it would be nice to split the sendLoss function into different functions, like sendLoss, sendMetrics.

Additional context
For cases when working with GANs, that we have two different models with different losses, or maybe even one single model with two weighted losses.

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions