@@ -0,0 +1,14 @@
+# Copyright (c) OpenMMLab. All rights reserved.
+from mmcv.parallel import is_module_wrapper
+from mmcv.runner import HOOKS, Hook
+
+
+@HOOKS.register_module()
+class SetEpochHook(Hook):
+
+    def before_train_epoch(self, runner):
+        epoch = runner.epoch
+        model = runner.model
+        if is_module_wrapper(model):
+            model = model.module
+        model.set_epoch(epoch)
1. SetEpochInfoHook 2. Add a short description/summary in docstring though it is simple