import gc
import torch


def empty_memory():
    """
    Performs garbage collection and empty cache in cuda device.
    """
    gc.collect()
    torch.cuda.empty_cache()


def tune_for_low_memory():
    """
    Tunes PyTorch to use float16 to reduce memory footprint.
    """
    torch.set_default_dtype(torch.float16)