OptimizerBase.ixx File Reference

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

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

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

Namespaces
namespace	Mila
	Mila main API namespace.

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 precisions.