search

FluxML / FluxJS.jl

I heard you like compile times
Other
42 stars 8 forks source link

compile function not working in case of Conv layers. #4

Open ayush1999 opened 6 years ago

ayush1999 commented 6 years ago

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.

MikeInnes commented 6 years ago

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