fix Flux compat

[mloubout]

Fixes #78

[codecov]

Codecov Report

Patch coverage: 92.85% and project coverage change: +0.06 🎉

Comparison is base (5c092f9) 88.48% compared to head (b2eeb12) 88.54%.

Additional details and impacted files

@@            Coverage Diff             @@
##           master      #79      +/-   ##
==========================================
+ Coverage   88.48%   88.54%   +0.06%     
==========================================
  Files          33       33              
  Lines        2545     2550       +5     
==========================================
+ Hits         2252     2258       +6     
+ Misses        293      292       -1     

Impacted Files	Coverage Δ
src/utils/chainrules.jl	92.15% <87.50%> (+0.66%)	⬆️
src/layers/invertible_layer_glow.jl	97.29% <100.00%> (ø)
src/layers/invertible_layer_hint.jl	94.39% <100.00%> (ø)
src/utils/dimensionality_operations.jl	95.69% <100.00%> (ø)
src/utils/neuralnet.jl	77.55% <100.00%> (+2.55%)	⬆️

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.

☔ View full report in Codecov by Sentry.
📢 Do you have feedback about the report comment? Let us know in this issue.