There is a newer version of the record available.

Published February 10, 2021 | Version v0.2.1
Software Open

NNGeometry: Easy and Fast Fisher Information Matrices and Neural Tangent Kernels in PyTorch

Authors/Creators

  • 1. Mila - Université de Montréal

Description

This minor releases makes sure that users are aware of the fact that we currently do not support batch norm in training mode. Prior releases would instead silently fail, without telling the user that the returned objects are likely wrong.

Files

tfjgeorge/nngeometry-v0.2.1.zip

Files (1.5 MB)

Name Size Download all
md5:37c5d6675812ca336aa11e2b711d908c
1.5 MB Preview Download

Additional details

Related works

Is supplement to
https://github.com/tfjgeorge/nngeometry/tree/v0.2.1 (URL)