Any ways to access the runner?

[twmht]

if i want to access the current epoch inside the model forwarding, i have to access the runner, but how can i do that?

[zhouzaida]

You can access the epoch from MessageHub instance which is a global object.

class CustomModule:
    def __init__(self):
        self.message_hub = MessageHub.get_current_instance()

    def forward(self, xxx):
        epoch = message_hub.get_info('epoch')

References:

• mmengine/mmengine/hooks/runtime_info_hook.py

  Line 77 in 5753cd9

  runner.message_hub.update_info('epoch', runner.epoch)

• https://mmengine.readthedocs.io/en/latest/design/logging.html#messagehub