issues
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
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Jaco revamp
#34
tfjgeorge
closed
2 years ago
1
Adds a very simple ResNet-like network to tests
#33
tfjgeorge
closed
2 years ago
1
Montecarlo Sampling Question
#32
simomagi
closed
2 years ago
1
adds a linear batchnorm layer in tests
#31
tfjgeorge
closed
2 years ago
1
Support of ParameterList
#30
arthurdouillard
closed
2 years ago
1
Implementing BatchNorm for KFAC
#29
a-cowlagi
closed
1 year ago
7
Extracting Eigenvalues of Fisher using KFAC Representation
#28
a-cowlagi
closed
1 year ago
1
switches to full backward hooks in jacobian generator
#27
tfjgeorge
closed
3 years ago
0
Pytorch Warning -- Non-Full Backward Hook when the Forward Contains Multiple Autograd Nodes
#26
a-cowlagi
closed
3 years ago
3
adds MonteCarlo Fisher helper for segmentation
#25
tfjgeorge
closed
3 years ago
0
compute_correlation() intent
#24
perone
closed
3 years ago
2
Support for ConvTranspose2d layers?
#23
ErikBertolino
closed
2 years ago
4
adds grad for PVectors
#22
tfjgeorge
closed
3 years ago
0
[Feature request] - multiclass semantic segmentation variant for FIM_MonteCarlo
#21
godaup
closed
3 years ago
3
adds size method to PVector
#20
tfjgeorge
closed
3 years ago
1
How is it possible to handle a model that has a BatchNorm layer using the PMatEKFAC representation to get the FIM ?
#19
johnrachwan123
closed
3 years ago
11
Adds pickle for PMat objects and PVectors
#18
tfjgeorge
closed
3 years ago
0
[Feature request] method to save/load FIM matrix efficiently
#17
TLESORT
closed
3 years ago
1
adds dot product to PVector
#16
tfjgeorge
closed
3 years ago
1
Update layercollection.py
#15
milpan
closed
2 years ago
4
updates torchvision for travis
#14
tfjgeorge
closed
3 years ago
0
fixes PVector bug when no bias, adds test
#13
tfjgeorge
closed
3 years ago
0
removes legacy function
#12
tfjgeorge
closed
3 years ago
0
fixes FIM MC for log_softmax
#11
tfjgeorge
closed
3 years ago
0
Mm diag dense
#10
tfjgeorge
closed
3 years ago
0
adds solve to lowrank repr
#9
tfjgeorge
closed
3 years ago
1
adds matrix-matrix product to PMatKFAC
#8
tfjgeorge
closed
3 years ago
1
adds add_to_model for PVector, fixes bug in copy_to_model
#7
tfjgeorge
closed
3 years ago
0
adds seed for deterministic tests
#6
tfjgeorge
closed
3 years ago
0
moves loader out of generator instance to be passed when calling meth…
#5
tfjgeorge
closed
3 years ago
0
An example for gram matrix computation?
#4
liutianlin0121
closed
3 years ago
2
Gn
#3
tfjgeorge
closed
4 years ago
0
Batch norm
#2
tfjgeorge
closed
4 years ago
0
adds kfac mv and vector representation including dict of mods and fla…
#1
tfjgeorge
closed
4 years ago
0
Previous