open-mmlab/mmengine

scattering the data to gpu when using base dataelement

Closed this issue · 1 comments

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

Hi, you can use the cuda method.

def cuda(self) -> 'BaseDataElement':