FluxJS.jl icon indicating copy to clipboard operation
FluxJS.jl copied to clipboard

compile function not working in case of Conv layers.

Open ayush1999 opened this issue 6 years ago • 1 comments

FluxJS.compile() doesn't seem to work when Convolution layers are involved:

>>> using Flux
>>> using FluxJS
>>> m = Chain(
         Conv((2,2), 1=>16, relu),
         x -> maxpool(x, (2,2)),
         Conv((2,2), 16=>8, relu),
         x -> maxpool(x, (2,2)),
         x -> reshape(x, :, size(x, 4)),
         Dense(288, 10), softmax)
>>> FLuxJS.compile("mnist",m, rand(10))
Error: StackOverflow

The error message is too big.

ayush1999 avatar Mar 10 '18 05:03 ayush1999

Yeah, we need a primitive definition that tells FluxJS how to map Julia convolutions to JS ones; something the the softmax one.

MikeInnes avatar Mar 12 '18 17:03 MikeInnes