PrepareTTAHook — mmengine 0.10.7 documentation (original) (raw)

• Docs >
• mmengine.hooks >
• PrepareTTAHook
• 以中文阅读

Shortcuts

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

Wraps runner.model with subclass of BaseTTAModel inbefore_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

• PrepareTTAHook
  • PrepareTTAHook
    * PrepareTTAHook.before_test()