CollosalAIModelWrapper¶
- class mmengine._strategy.colossalai.CollosalAIModelWrapper(model_wrapper, model)[source]¶
- Parameters
model_wrapper (None) –
model (torch.nn.modules.module.Module) –
- test_step(data)¶
Gets the prediction of module during validation process.