PrepareTTAHook

class mmengine.hooks.PrepareTTAHook(tta_cfg)

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)

Wraps runner.model with the subclass of BaseTTAModel.

Parameters

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

Return type

None