MMDeepSpeedEngineWrapper¶ class mmengine._strategy.deepspeed.MMDeepSpeedEngineWrapper(*, model, inputs_to_half=None)[源代码]¶ 参数: model (deepspeed.DeepSpeedEngine) – inputs_to_half (List[int | str] | None) – test_step(data)[源代码]¶ Gets the predictions of module during testing process. 参数: data (dict or tuple or list) – Data sampled from dataset. 返回: The predictions of given data. 返回类型: list val_step(data)[源代码]¶ Gets the prediction of module during validation process. 参数: data (dict or tuple or list) – Data sampled from dataset. 返回: The predictions of given data. 返回类型: list