search

alan-turing-institute / ThermodynamicAnalyticsToolkit

Sampling-based approach to analyse neural networks using TensorFlow
https://alan-turing-institute.github.io/ThermodynamicAnalyticsToolkit/
GNU General Public License v3.0
22 stars 3 forks source link

Add diagonal hessians #37

Open FrederikHeber opened 6 years ago

FrederikHeber commented 6 years ago

Gabriel suggested to add an option to only evaluate the diagonal hessians instead of the full hessian matrix.