docs: keep the ConvMixer default backend as cuda.jl for now

LuxDL · Nov 11, 2024 · 43ac5ef · 43ac5ef
1 parent cf05b82
commit 43ac5ef
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/ConvMixer/main.jl b/examples/ConvMixer/main.jl
@@ -76,7 +76,7 @@ end
 Comonicon.@main function main(; batchsize::Int=512, hidden_dim::Int=256, depth::Int=8,
         patch_size::Int=2, kernel_size::Int=5, weight_decay::Float64=1e-5,
         clip_norm::Bool=false, seed::Int=42, epochs::Int=25, lr_max::Float64=0.01,
-        backend::String="reactant")
+        backend::String="gpu_if_available")
     rng = StableRNG(seed)
 
     if backend == "gpu_if_available"

diff --git a/src/layers/pooling.jl b/src/layers/pooling.jl
@@ -14,7 +14,7 @@ end
 struct GlobalPoolMode <: AbstractPoolMode end
 
 # XXX: Use PoolDims once EnzymeJAX supports stablehlo.reduce_window adjoint
-(::GlobalPoolMode)() = GlobalPoolMode()
+(g::GlobalPoolMode)(::Any) = g
 
 @concrete struct AdaptivePoolMode <: AbstractPoolMode
     out_size <: Tuple{Vararg{IntegerType}}