mmedit.engine.runner.multi_loops

Module Contents

Classes

MultiValLoop	Loop for validation multi-datasets.
MultiTestLoop	Loop for validation multi-datasets.

class mmedit.engine.runner.multi_loops.MultiValLoop(runner, dataloader: Union[torch.utils.data.DataLoader, Dict], evaluator: Union[mmengine.evaluator.Evaluator, Dict, List], fp16: bool = False)

Bases: mmengine.runner.base_loop.BaseLoop

Loop for validation multi-datasets.

参数

• runner (Runner) – A reference of runner.

• dataloader (list[Dataloader or dic]) – A dataloader object or a dict to build a dataloader.

• evaluator (list[]) – Used for computing metrics.

• fp16 (bool) – Whether to enable fp16 validation. Defaults to False.

run()

Launch validation.

run_iter(idx: int, data_batch: Sequence[dict])

Iterate one mini-batch.

参数

• idx (int) – The index of the current batch in the loop.

• data_batch (Sequence[dict]) – Batch of data from dataloader.

class mmedit.engine.runner.multi_loops.MultiTestLoop(runner, dataloader: Union[torch.utils.data.DataLoader, Dict], evaluator: Union[mmengine.evaluator.Evaluator, Dict, List], fp16: bool = False)

Bases: mmengine.runner.base_loop.BaseLoop

Loop for validation multi-datasets.

参数

• runner (Runner) – A reference of runner.

• dataloader (Dataloader or dict) – A dataloader object or a dict to build a dataloader.

• evaluator (Evaluator or dict or list) – Used for computing metrics.

• fp16 (bool) – Whether to enable fp16 validation. Defaults to False.

run()

Launch test.

run_iter(idx: int, data_batch: Sequence[dict])

Iterate one mini-batch.

参数

• idx (int) – The index of the current batch in the loop.

• data_batch (Sequence[dict]) – Batch of data from dataloader.