megengine.dtr

eviction_threshold

Get or set the eviction threshold in bytes.

evictee_minimum_size

Get or set the memory threshold of tensors in bytes.

enable_sqrt_sampling

Get or set whether sqrt sampling is allowed.

enable

启用记录 Tensor 的计算路径并执行 DTR 策略。

disable

停止记录 Tensor 的计算路径和执行 DTR 策略。