Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision > Member List

This is the complete list of members for Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >, including all inherited members.

AdamWOptimizer(IExecutionContext *exec_context, const AdamWConfig &config)Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >inlineexplicit
addParameter(ITensor *param, ITensor *grad) overrideMila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >inlinevirtual
ExecutionContextType typedef (defined in Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >)Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >
getBeta1() const noexcept (defined in Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >)Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >inline
getBeta2() const noexcept (defined in Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >)Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >inline
getEpsilon() const noexcept (defined in Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >)Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >inline
getLearningRate() const overrideMila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >inlinevirtual
getParameterCount() const noexcept (defined in Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >)Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >inline
getStepCount() const noexcept (defined in Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >)Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >inline
getWeightDecay() const noexcept (defined in Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >)Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >inline
OptimizerType typedef (defined in Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >)Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >
setLearningRate(float learning_rate) overrideMila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >inlinevirtual
setWeightDecay(float weight_decay) (defined in Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >)Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >inline
step() overrideMila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >inlinevirtual
~AdamWOptimizer() override=default (defined in Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >)Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >
~Optimizer()=default (defined in Mila::Dnn::Optimizer< TDeviceType, TPrecision >)Mila::Dnn::Optimizer< TDeviceType, TPrecision >virtual