mmagic.engine.runner.log_processor

Module Contents

Classes

LogProcessor

LogProcessor inherits from mmengine.runner.LogProcessor and

class mmagic.engine.runner.log_processor.LogProcessor(window_size=10, by_epoch=True, custom_cfg: Optional[List[dict]] = None, num_digits: int = 4, log_with_hierarchy: bool = False, mean_pattern='.*(loss|time|data_time|grad_norm).*')

Bases: mmengine.runner.LogProcessor

LogProcessor inherits from mmengine.runner.LogProcessor and overwrites self.get_log_after_iter().

This log processor should be used along with mmagic.engine.runner.MultiValLoop and mmagic.engine.runner.MultiTestLoop.

_get_dataloader_size(runner, mode) → int

Get dataloader size of current loop. In MultiValLoop and MultiTestLoop, we use total_length instead of len(dataloader) to denote the total number of iterations.

Parameters

• runner (Runner) – The runner of the training/validation/testing

• mode (str) – Current mode of runner.

Returns

The dataloader size of current loop.

Return type

int