MMDeepSpeedEngineWrapper¶ class mmengine._strategy.deepspeed.MMDeepSpeedEngineWrapper(*, model, inputs_to_half=None)[source]¶ Parameters model (deepspeed.DeepSpeedEngine) – inputs_to_half (Optional[List[Union[int, str]]]) – test_step(data)[source]¶ Gets the predictions of module during testing process. Parameters data (dict or tuple or list) – Data sampled from dataset. Returns The predictions of given data. Return type list val_step(data)[source]¶ Gets the prediction of module during validation process. Parameters data (dict or tuple or list) – Data sampled from dataset. Returns The predictions of given data. Return type list