RD-Agent
RD-Agent copied to clipboard
Add GPU support for time series models in Co-STEER (fixes #1256)
GPU Support for Time Series Models
Resolves #1256
Problem
Time series models (LSTM) in Co-STEER framework were CPU-only, causing significant resource waste.
Solution
Implement automatic GPU detection and acceleration while maintaining full CPU compatibility.
Changes
rdagent/utils/gpu_utils.py- GPU detection & utilitiesrdagent/general_model/general_model.py- Enhanced LSTM modelsrdagent/data_science/loop.py- GPU-optimized trainingrdagent/core/evolving_framework.py- Co-STEER integrationtest/utils/test_gpu_support.py- Comprehensive testing
📚 Documentation preview 📚: https://RDAgent--1264.org.readthedocs.build/en/1264/