Flux CNN error: Scalar indexing is disallowed

Specific Domains Machine Learning
, ,
2 
 [11] conv(x::CuArray{Int64, 4, CUDA.Mem.DeviceBuffer}, w::CuArray{Float32, 4, CUDA.Mem.DeviceBuffer}, cdims::DenseConvDims{2, 2, 2, 4, 2}; kwargs::Base.Iterators.Pairs{Union{}, Union{}, Tuple{}, NamedTuple{(), Tuple{}}})
    @ NNlib /user/jlib/packages/NNlib/0QnJJ/src/conv.jl:88

Something is producing an array of integers x here, maybe your input data? The GPU conv wants all Float32, so instead you get some slow fallback, which works by indexing.

1 Like