RuntimeInfoHook¶
- class mmengine.hooks.RuntimeInfoHook[source]¶
A hook that updates runtime information into message hub.
E.g.
epoch
,iter
,max_epochs
, andmax_iters
for the training state. Components that cannot access the runner can get runtime information through the message hub.- after_test_epoch(runner, metrics=None)[source]¶
All subclasses should override this method, if they need any operations after each test epoch.
- after_train_iter(runner, batch_idx, data_batch=None, outputs=None)[source]¶
Update
log_vars
in model outputs every iteration.
- after_val_epoch(runner, metrics=None)[source]¶
All subclasses should override this method, if they need any operations after each validation epoch.