Error

Prevent `CUDA error: out of memory` in just 1 line of code

Koila solves CUDA error: out of memory error painlessly.
Fix it with just one line of code, and forget it.

Type Checking
Formatting
Unit testing
License: MIT
Tweet

Koila

🚀 Features

  • 🙅 Prevents CUDA error: out of memory error with one single line of code.

  • 🦥 Lazily evaluates pytorch code to save computing power.

  • ✂️ Automatically splits along the batch dimension to more GPU friendly numbers (2’s powers) to speed up the execution.

  • 🤏

     

     

     

    To finish reading, please visit source site