Class AdamW
java.lang.Object
io.github.kirstenali.deepj.optimisers.AdamW
- All Implemented Interfaces:
ParameterOptimizer
AdamW optimizer with per-parameter state.
State is keyed by Parameter identity, and updates are applied in-place.
-
Constructor Summary
ConstructorsConstructorDescriptionAdamW(double lr, double beta1, double beta2, double eps, double weightDecay) -
Method Summary
Modifier and TypeMethodDescriptionstatic AdamWdefaultAdamW(double lr) voidUpdate the provided parameters (one optimizer step).
-
Constructor Details
-
AdamW
public AdamW(double lr, double beta1, double beta2, double eps, double weightDecay)
-
-
Method Details
-
defaultAdamW
-
step
Description copied from interface:ParameterOptimizerUpdate the provided parameters (one optimizer step).- Specified by:
stepin interfaceParameterOptimizer
-