Function eviction_threshold

imperative/python/megengine/dtr/dtr.py:25–41

r"""Get or set the eviction threshold in bytes. It can also be set to a string, whose formatting supports byte(B), kilobyte(KB), megabyte(MB) and gigabyte(GB) units. Note: When GPU memory usage exceeds this value, DTR will heuristically select and evict resident tenso

(mod)

@property
def eviction_threshold(mod):
    r"""Get or set the eviction threshold in bytes. It can also be set to a string,
    whose formatting supports byte(B), kilobyte(KB), megabyte(MB) and
    gigabyte(GB) units.
    
    Note: 
       When GPU memory usage exceeds this value, DTR will heuristically select
       and evict resident tensors until the amount of used memory falls below
       this threshold.
    
    Examples:
        .. code-block::

           import megengine as mge
           mge.dtr.eviction_threshold = "2GB"
    """
    return _eviction_threshold


@eviction_threshold.setter