MMDeepSpeedEngineWrapper

class mmengine._strategy.deepspeed.MMDeepSpeedEngineWrapper(*, model, inputs_to_half=None)

参数

• model (deepspeed.DeepSpeedEngine) –

• inputs_to_half (Optional[List[Union[int, str]]]) –

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