Flux relu fails with CUDA gpu

I’m getting error “CuArray only supports bits types” when using relu in code like:

model = Chain(Dense(2, 2, relu), Dense(2, 1, identity)) |> gpu

All works fine though with:

model = Chain(Dense(2, 2, (x) -> max(0, x), Dense(2, 1, identity)) |> gpu

Can anyone help how to fix relu or I should use below code?

3 posts - 2 participants

Read full topic