|
Mila 0.13.48
Deep Neural Network Library
|
AdamW optimizer wrapper using fluent AdamWConfig. More...
#include <memory>#include <type_traits>#include <stdexcept>import Compute.CpuAdamWOptimizer;import Compute.DeviceType;import Compute.OptimizerBase;import Compute.ExecutionContext;import Dnn.TensorDataType;import Dnn.TensorTypes;import Dnn.TensorDataTypeTraits;import Dnn.ITensor;import Dnn.Optimizers.AdamWConfig;Classes | |
| class | Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision > |
| Device-agnostic AdamW optimizer. More... | |
Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
| namespace | Mila::Dnn |
| namespace | Mila::Dnn::Optimizers |
AdamW optimizer wrapper using fluent AdamWConfig.