Update utils.jl

YichengDWu · Oct 21, 2022 · f09ed3b · f09ed3b
1 parent 0bbd77a
commit f09ed3b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/utils.jl b/src/utils.jl
@@ -90,4 +90,4 @@ function isongpu(nt::NamedTuple)
     return any(x -> x isa AbstractGPUArray, Lux.fcollect(nt))
 end
 
-float64 = Base.Fix1(Broadcast.broadcast, Float64)
+float64 = Base.Fix1(convert, AbstractArray{Float64})