ColossalAI
ColossalAI copied to clipboard
[BUG]:
🐛 Describe the bug
- Get cuda_emptyCache when use Pyrtorch lightning ColossalAIStrategy(https://github.com/Lightning-AI/lightning/blob/master/src/pytorch_lightning/strategies/colossalai.py)

Environment
No response