swalot 0.0.1

No CUDA out of memory again; fully use of a GPU card per user.

Installation

You can install the Memory Protector package using pip:

pip install swalot

Usage

Simply import and wrap training code:

import swalot as sw

with sw.protext():
    """
    use CUDA tensor calculation here as usual.
    All RAM will be protected automatically!
    e.g.
    """
    # a = torch.randn(1000, 1000, 600).cuda()