PPO 通过修改配置降低显存占用

PaddlePaddle / PARL

A high-performance distributed training framework for Reinforcement Learning

https://parl.readthedocs.io/

Apache License 2.0

3.24k stars 819 forks source link

Closed Aganlengzi closed 2 years ago

Aganlengzi commented 2 years ago

请问，https://github.com/PaddlePaddle/PARL/tree/develop/examples/PPO 是否可以通过修改配置降低显存占用？在一个芯片上单卡跑到Updates 400+ 时报out of memory了

TomorrowIsAnOtherDay commented 2 years ago

把batch_size调小一点？

Aganlengzi commented 2 years ago

多谢回复，已调到16训完

TomorrowIsAnOtherDay commented 2 years ago

可以star下PARL关注我们后续的研发：）