EmptyCacheHook¶ 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.