Skip to content

Model weights and gradients in TrainKeras #1988

Open

Description

@brollb

It would be nice to visualize the model weights and gradients in TrainKeras as this can be useful in detecting things like vanishing/exploding gradients.

Activity

  1. brollb commented on Jan 13, 2021

    @brollb
    ContributorAuthor

    Should these be visualized on the training architecture?
    The approach used with pipelines/executions could be useful here, too. The neural network architectures could be augmented with "execution" information such as model weights which can then be visualized... Maybe this would be best to add as metadata to trained Keras models which can be used to view model checkpoints.

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

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