Skip to content

Tensorboard-logging, gradient visualisation, bugfixes

Compare
Choose a tag to compare
@StillerPatrick StillerPatrick released this 28 Oct 13:12
· 63 commits to master since this release
cc62b5a

New Features

Tensorboard logging is available and can be activated in the fit() function. Logger enables automatic tracking of all loss terms and their weights

Logging of Loss gradients in order to identity gradient pathologies, you can track the gradient updates by adding a logger and setting the
track_gradient flag in the fit() function.

Bugfixes

Fix the usage of the to method in MLP and FingerNet

Examples

A working example of the heat equation