megengine.dtr.enable_sqrt_sampling

enable_sqrt_sampling = <property object>

获取或设置是否允许 sqrt 采样。 sqrt 采样意味着给定候选集的大小为 N，枚举 sqrt(N)。 当 Tensor 的数量非常多时，启用此优化将加快训练速度。

实际案例

import megengine as mge
mge.dtr.enable_sqrt_sampling = True

上一页

megengine.dtr.evictee_minimum_size

下一页

megengine.dtr.enable