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