Closed torfjelde closed 3 years ago
Simplified Logit implementation a bit and fixed constructor. Currently constructors such as Logit{1}(0, 1) are not supported, and this PR fixes this.
Logit{1}(0, 1)
I just need to add some tests for the constructor and this should be good to go.
Zygote test errors should be fixed by https://github.com/FluxML/Zygote.jl/pull/934.
Seems like it's good to go?
Simplified Logit implementation a bit and fixed constructor. Currently constructors such as
Logit{1}(0, 1)
are not supported, and this PR fixes this.I just need to add some tests for the constructor and this should be good to go.