CarperAI / DRLX

Diffusion Reinforcement Learning Library
MIT License
171 stars 7 forks source link

performance optimizations #7

Closed tmabraham closed 1 year ago

tmabraham commented 1 year ago

with everything enabled (tf32, gradient checkpointing, attention slicing, xformers memory-efficient attention), speeds up single GPU performance by pretty much 2x.