Shortcuts

PrepareTTAHook

class mmengine.hooks.PrepareTTAHook(tta_cfg)[source]

Wraps runner.model with subclass of BaseTTAModel in before_test.

Note

This function will only be used with MMFullyShardedDataParallel.

Parameters

tta_cfg (dict) – Config dictionary of the test time augmentation model.

before_test(runner)[source]

Wraps runner.model with the subclass of BaseTTAModel.

Parameters

runner (Runner) – The runner of the testing process.

Return type

None

Read the Docs v: v0.5.0
Versions
latest
stable
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.