/GFdynamics

Code used for "Gradient flow dynamics of shallow ReLU networks for square loss and orthogonal inputs"

Primary LanguageJupyter Notebook

Watchers