How could I find the current epoch number in the code?

[EdyQiu]

Hi, I want to add an auxiliary loss in the head module, and it is designed to start working when the epoch number > 5.
Therefore I want to ask how could I know the current epoch number, is there any function for that?
Thanks for help!

[Ezra-Yu]

Based on your needs, it is highly recommended that you use SwitchRecipeHook.

loss_after_30e = [dict(type='CE'), dict(type='Mine')]
loss_after_60e = [dict(type='CE')]
custom_hooks = [
    dict(
        type='SwitchRecipeHook',
        schedule=[
            dict(action_epoch=30, loss=loss_after_30e),
            dict(action_epoch=60, loss=loss_after_60e),
        ]
    )
]