fix: incorrect activation usage

LuxDL · Jul 22, 2024 · 575cfa4 · 575cfa4
1 parent c3052eb
commit 575cfa4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/impl/bias_activation.jl b/src/impl/bias_activation.jl
@@ -142,7 +142,7 @@ function __apply_bias_activation_cached!!(
     if can_setindex(x)
         opmode = internal_operation_mode((x, bias))
         if opmode isa LoopedArrayOp
-            bc = Broadcast.instantiate(Broadcast.broadcasted(σ ∘ +, x, bias_))
+            bc = Broadcast.instantiate(Broadcast.broadcasted(+, x, bias_))
             @simd ivdep for I in eachindex(bc)
                 @inbounds x[I] = bc[I]
             end