https://github.com/rentruewang/koila

Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.
https://github.com/rentruewang/koila

deep-learning gradient-accumulation lazy-evaluation machine-learning memory-management neural-network out-of-memory python pytorch

Last synced: 5 months ago
JSON representation

Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.

• Host: GitHub
• URL: https://github.com/rentruewang/koila
• Owner: rentruewang
• License: mit
• Created: 2021-11-17T08:37:36.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2024-11-03T18:15:42.000Z (5 months ago)
• Last Synced: 2024-11-11T06:42:52.773Z (5 months ago)
• Topics: deep-learning, gradient-accumulation, lazy-evaluation, machine-learning, memory-management, neural-network, out-of-memory, python, pytorch
• Language: Python
• Homepage: https://koila.rentruewang.com
• Size: 3.72 MB
• Stars: 1,821
• Watchers: 12
• Forks: 63
• Open Issues: 12
• Metadata Files:
  • Readme: README.md
  • License: LICENSE.md
  • Code of conduct: CODE_OF_CONDUCT.md

Awesome Lists containing this project

• Awesome-pytorch-list - Koila