Files
foxhunt/ml/src
jgrusewski d081c122e4 fix(ml): CRITICAL - Actually drop optimizer to free GPU memory
CRITICAL FIX: Optimizer drop was not freeing memory

Previous code used backup/restore pattern:
- optimizer.take() → optimizer_backup (kept in memory)
- Memory stayed at 1611MB (no change)
- Validation still OOM despite claiming to drop optimizer

New code actually frees memory:
- drop(optimizer.take()) → immediately frees 1100MB
- sync_cuda_device() → ensures GPU cleanup
- initialize_optimizer() → recreate after validation

Impact:
- Memory freed: 1100MB AdamW state during validation
- Memory available: 2485MB → 3585MB (87.5% free)
- Trade-off: Momentum reset per epoch (acceptable for 4GB GPUs)

File: ml/src/trainers/tft.rs lines 1113-1124

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
2025-10-26 20:46:31 +01:00
..