EmptyCacheHook — mmengine 0.10.7 documentation (original) (raw)

• Docs >
• mmengine.hooks >
• EmptyCacheHook
• 以中文阅读

Shortcuts

class mmengine.hooks.EmptyCacheHook(before_epoch=False, after_epoch=True, after_iter=False)[source]¶

Releases all unoccupied cached GPU memory during the process of training.

Parameters:

• before_epoch (bool) – Whether to release cache before an epoch. Defaults to False.

• after_epoch (bool) – Whether to release cache after an epoch. Defaults to True.

• after_iter (bool) – Whether to release cache after an iteration. Defaults to False.

• EmptyCacheHook

  • EmptyCacheHook