search

slimgroup / InvertibleNetworks.jl

A Julia framework for invertible neural networks
MIT License
148 stars 20 forks source link

log of determinant for inverse pass #70

Open kaido975 opened 1 year ago

kaido975 commented 1 year ago

If I am not missing it, It would be great to have this feature.

rafaelorozco commented 1 year ago

Yes I agree, I have a branch with this implemented. I will open a PR later today. Will you mostly be using the GLOW network?

rafaelorozco commented 1 year ago

Ok, this branch https://github.com/slimgroup/InvertibleNetworks.jl/tree/glow-map implements the logdet on inverse and the backward_inv for glow network. The code base doesn't expect inverse to output logdet so some of the other network tests fail. But I made sure that the glow network and layers tests pass. I will fix the rest of the tests to merge to master in a bit but you are welcome to play around with that branch in the meanwhile.

fabienbaron commented 3 months ago

Looks like this has not been merged? Useful though.