Shortcuts

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.

Read the Docs v: latest
Versions
master
latest
stable
zyh-doc-notfound-extend
Downloads
pdf
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.