def flush()

in utils/benchmarking_utils.py [0:0]

• 5 lines of code
• 1 McCabe index (conditional complexity)

def flush():
    """Wipes off memory."""
    gc.collect()
    torch.cuda.empty_cache()
    torch.cuda.reset_max_memory_allocated()
    torch.cuda.reset_peak_memory_stats()