We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e03213c commit 34a964eCopy full SHA for 34a964e
src/flows/realnvp.jl
@@ -152,7 +152,7 @@ function RealNVP(
152
create_flow(Ls, q0)
153
end
154
155
-function RealNVP(dims:Int; paramtype::Type{T} = Float64) where {T<:AbstractFloat}
+function RealNVP(dims::Int; paramtype::Type{T} = Float64) where {T<:AbstractFloat}
156
# default RealNVP with 10 layers, each couplling function has 2 hidden layers with 32 units
157
return RealNVP(dims, [32, 32], 10; paramtype=paramtype)
158
0 commit comments