CollosalAIModelWrapper

class mmengine._strategy.colossalai.CollosalAIModelWrapper(model_wrapper, model)

参数

• model_wrapper (None) –

• model (torch.nn.modules.module.Module) –

test_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

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