Flux.jl: training fails at GPU but works on CPU

hmmm, looks like somewhere in my file iterations the definition of CuArrays.culiteral_pow(::typeof(^), x::ForwardDiff.Dual{Nothing,Float32,1}, ::Val{2}) got accidentally changed… It should be x*x, and not x, of course :slight_smile:

1 Like