ContinualAI / ContinualAI/avalanche-rl
Memory Management
Open
bug
- Dominant language
- Python
- Stars
- 85
- Forks
- 15
- PR merge metrics
- No merged PRs in 30d
Description
Hello,
In avalanhce-rl, after 1000 experiences and 1000 steps per experience, the below error will be risen. It is necessary to manage memory during training correctly.
torch.cuda.OutOfMemoryError
/Avalanche/avalanche-rl/avalanche_rl/training/strategies/rl_base_strategy.py", line 377, in train_exp
self.loss.backward()
Thanks.
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.