search

deepforge-dev / deepforge

A modern development environment for deep learning
https://deepforge.org
Apache License 2.0
754 stars 78 forks source link

Model weights and gradients in TrainKeras #1988

Open brollb opened 3 years ago

brollb commented 3 years ago

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.

brollb commented 3 years ago

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.