PrepareTTAHook

class mmengine.hooks.PrepareTTAHook(tta_cfg)

Wraps runner.model with subclass of BaseTTAModel in before_test.

备注

This function will only be used with MMFullyShardedDataParallel.

参数

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

before_test(runner)

Wraps runner.model with the subclass of BaseTTAModel.

参数

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

返回类型

None