MADE #39
MADE #39
Conversation
MADE/made.jl
Outdated
| rng = MersenneTwister(made.seed) | ||
| made.seed = (made.seed + 1) % made.num_masks | ||
|
|
||
| # sample the order of the inpdimensionsuts and the connectivity of all neurons |
MADE/made.jl
Outdated
| #Getting data. The data used here is binarized MNIST dataset | ||
|
|
||
| X = npzread("/path/to/your/data.npy") | ||
| X = X' |
There was a problem hiding this comment.
Punning on adjoint is not really recommended, right? And deprecated in 0.7, so maybe use permutedims?
| hs = push!([in], hs...) | ||
| layers = [MaskedDense(hs[i], hs[i + 1]; σ = relu) for i = 1:length(hs) - 1] | ||
|
|
||
| net = Chain(layers..., MaskedDense(hs[end], out)) |
There was a problem hiding this comment.
Couldn't this have a special case for the first and last layers (rather than just the last one), allowing you to avoid the push! in line 57?
There was a problem hiding this comment.
Wouldn't it be an issue if hs = []? In that case, Chain(MaskedDense(in, hs[1]), layers..., MaskedDense(hs[end], out)) gives BoundsError
MADE/made.jl
Outdated
|
|
||
| function set_mask(a::MaskedDense, mask) | ||
| a.mask = mask | ||
| end |
There was a problem hiding this comment.
Unless there's validation that you need to do this probably doesn't need to be a setter method and could just be replaced at it so call locations by directly setting the attribute?
| end | ||
|
|
||
| function MaskedDense(in::Integer, out::Integer; σ = identity) | ||
| return MaskedDense(param(glorot_uniform(out, in)), param(zeros(out)), ones(out, in), σ) |
There was a problem hiding this comment.
This could probably be modified to be more type-generic? Would be nice to be able to support GPU and/or Float32
There was a problem hiding this comment.
Could you please explain a bit more on that? Because in and out have to be integers, MaskedDense has been modeled after Dense.
MADE/made.jl
Outdated
| nin::Integer | ||
| nout::Integer | ||
| hidden_sizes::Array{Integer, 1} | ||
| net::Chain |
There was a problem hiding this comment.
These should probably be concretely typed (Int and either Vector{Int} or a tuple)
| L = length(made.hidden_sizes) | ||
|
|
||
| # fetch the next seed and construct a random stream | ||
| rng = MersenneTwister(made.seed) |
There was a problem hiding this comment.
Is this reinitialization going to be slow?
MADE/made.jl
Outdated
| num_masks | ||
| seed::UInt # for cycling through num_masks orderings | ||
|
|
||
| m::Dict |
There was a problem hiding this comment.
Maybe give this dict a concrete type for clarity about what it's going to hold?
|
Can you update this to use Also this line seemed sketchy, given that maskeddense is immutable, what's going on there? |
|
I have updated the model with MNIST, rather than binarized MNIST which was used by the original paper. I have also fixed the mentioned line. Its just that masks are getting updated there, similar to how the weights are updated after running through the optimizer (someone please correct me if I am wrong there). |
MikeInnes
force-pushed
the
master
branch
2 times, most recently
from
d7061a7 to
98fe429
Compare
MADE is Masked AutoEncoder for Density Estimation. @MikeInnes @jekbradbury