chaiyujin / glow-pytorch

pytorch implementation of openai paper "Glow: Generative Flow with Invertible 1×1 Convolutions"
MIT License
505 stars 79 forks source link

Fixed un-computable gradients #6

Closed kolchinski closed 5 years ago

kolchinski commented 5 years ago

Replaced in-place operations in coupling layers to resolve issues with gradients being unable to propagate through in-place ops