And changing the changed name in the tests.

FluxML · Dec 16, 2024 · b95fd3c · b95fd3c
1 parent c75142f
commit b95fd3c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/rules.jl b/test/rules.jl
@@ -14,7 +14,7 @@ RULES = [
   OptimiserChain(ClipNorm(), Adam(0.001)),
   OptimiserChain(ClipGrad(0.5), Momentum()),
   OptimiserChain(WeightDecay(), OAdam(), ClipGrad(1)),
-  OptimiserChain(GradNormGrowthLimiter(1.1), Apollo()),
+  OptimiserChain(NormGrowthCap(1.1), Apollo()),
   # Not the default:
   RMSProp(centred = true), AdamW(couple=false),
 ]