accelerate
b028a198 - Add a memory-aware decorator for CUDA OOM avoidance (#324)

Commit

3 years ago

Add a memory-aware decorator for CUDA OOM avoidance (#324) Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

References

Author

muellerzr

Parents