keep duplicate layers in a model

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

Closed toontran closed 1 year ago

toontran commented 1 year ago

Fixes #60

Sorry for the accidental PR earlier

tfjgeorge commented 1 year ago

Thanks for the PR but as discussed in #60 it is not the expected behavior. I am closing the PR.