search

tfjgeorge / nngeometry

{KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch
https://nngeometry.readthedocs.io
MIT License
203 stars 20 forks source link

keep duplicate layers in a model #61

Closed toontran closed 1 year ago

toontran commented 1 year ago

Fixes #60