Hyperbolic network now uses GPU supported Haar wavelet

slimgroup / InvertibleNetworks.jl

A Julia framework for invertible neural networks

MIT License

152 stars 23 forks source link

Closed PetersBas closed 3 years ago

PetersBas commented 3 years ago

Hyperbolic network now uses custom implementation of the (inverse) Haar transform with GPU support.
Hyperbolic network can no longer use the wavelets implemented using wavelet_squeeze form the wavelets.jl package (that did not support GPU)