Commit Graph
1 Commits
Author SHA1 Message Date
Ashar edc3b12345 fix(ara): free gradient buffers after optimization (#426)
LBFGS leaves one full-size gradient buffer on every optimized weight.
Across the layers processed in a typical trial this is many GiB of VRAM
that persists into evaluation, causing CUDA out-of-memory errors.
Clear the gradients after each module is optimized.
2026-08-17 14:57:09 +05:30