Merge branch 'main' into ap/fix_pool_debug

LuxDL · Dec 3, 2024 · d5b3e56 · d5b3e56
2 parents 0f14fb9 + 277f6ab
commit d5b3e56
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/src/api/Lux/utilities.md b/docs/src/api/Lux/utilities.md
@@ -13,6 +13,7 @@ basic building blocks which can be seamlessly composed to create complex trainin
 
 ```@docs
 Training.TrainState
+Training.TrainState(::AbstractLuxLayer, ::Any, ::Any, ::Optimisers.AbstractRule)
 Training.compute_gradients
 Training.apply_gradients
 Training.apply_gradients!