accelerate
a24189db - reserve 10% GPU in `get_balanced_memory` to avoid OOM (#1798)

Commit
2 years ago
reserve 10% GPU in `get_balanced_memory` to avoid OOM (#1798) * reserve 10% GPU to avoid OOM * update warning message Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> * use logger.info * clean up comment --------- Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Author
Parents
Loading