OptimizerBase.ixx File Reference

Base interface for neural network parameter optimizers. More...

#include <memory>
import Compute.DeviceType;
import Dnn.TensorDataTypeTraits;
import Dnn.TensorTypes;
import Dnn.TensorDataType;
import Dnn.ITensor;

Classes
class	Mila::Dnn::Compute::Optimizer< TDeviceType, TPrecision >
	Abstract base class for parameter optimizers. More...

Namespaces
namespace	Mila
	Mila main API namespace.
namespace	Mila::Dnn
namespace	Mila::Dnn::Compute

Detailed Description

Base interface for neural network parameter optimizers.

Defines the abstract Optimizer class template that provides a uniform interface for parameter update algorithms (SGD, Adam, AdamW, etc.) across different devices and precision levels.