megengine.dtr¶
Get or set the eviction threshold in bytes. |
|
Get or set the memory threshold of tensors in bytes. |
|
Get or set whether sqrt sampling is allowed. |
|
启用记录 Tensor 的计算路径并执行 DTR 策略。 |
|
停止记录 Tensor 的计算路径和执行 DTR 策略。 |
Get or set the eviction threshold in bytes. |
|
Get or set the memory threshold of tensors in bytes. |
|
Get or set whether sqrt sampling is allowed. |
|
启用记录 Tensor 的计算路径并执行 DTR 策略。 |
|
停止记录 Tensor 的计算路径和执行 DTR 策略。 |