draugr.torch_utilities.optimisation.debugging.gradients.flow

Created on 29/07/2020

Functions

plot_grad_flow(model[, lines, alpha, line_width])

Plots the gradients flowing through different layers in the net during training.