scripts/clear_memory.py

import gc
import torch
from logger import rich_logger as l

def clear_memory():
    """
    Clears the memory by collecting garbage and emptying the CUDA cache.

    This function is useful when dealing with memory-intensive operations in Python, especially when using libraries like PyTorch.

    Note:
        This function requires the `gc` and `torch` modules to be imported.

    """
    gc.collect()
    torch.cuda.empty_cache()
    l.info("Memory Cleared")