dynamic-batch-sizing

Tag

Cards List
#dynamic-batch-sizing

Prevent CUDA OOM in PyTorch with dynamic lane switching

Reddit r/LocalLLaMA ↗ · yesterday

MEM v3 is a memory management tool for PyTorch that dynamically adjusts batch size and gradient accumulation to prevent CUDA OOM crashes during training and fine-tuning, featuring chaos resistance, crash-proof checkpoints, and live telemetry.

0 favorites 0 likes
← Back to home

Submit Feedback