ClearCudaCache

class torchrl.trainers.ClearCudaCache(interval: int)

以给定间隔清除 cuda 缓存。

示例

>>> clear_cuda = ClearCudaCache(100)
>>> trainer.register_op("pre_optim_steps", clear_cuda)

abstract register(trainer: Trainer, name: str)

在训练器中的默认位置注册钩子。

参数：

• trainer (Trainer) – 必须注册钩子的训练器。

• name (str) – 钩子的名称。

注意

要在默认位置以外的其他位置注册钩子，请使用 register_op()。