open-mmlab/mmengine

scattering the data to gpu when using base dataelement

Closed this issue · 1 comments

ajaynitk commented

What is the feature?

It is convenient if we have support for scattering data to gpus's when using base data element like we have for data container in the previous open mmlab versions.

Any other context?

No response

zhouzaida commented

Hi, you can use the cuda method.

mmengine/mmengine/structures/base_data_element.py

Line 503 in 9b98405

def cuda(self) -> 'BaseDataElement':