Namespace AiDotNet.Training.Memory

Classes

MemorySavingsEstimate

Memory savings estimate from optimization techniques.

TrainingMemoryConfig

Configuration for training memory management including gradient checkpointing, activation pooling, and model sharding.

TrainingMemoryManager<T>

Manages memory optimization during neural network training including gradient checkpointing, activation pooling, and model sharding.