diffengine.models.editors.esd.esd_xl_data_preprocessor

Module Contents

Classes

ESDXLDataPreprocessor

ESDXLDataPreprocessor.
class diffengine.models.editors.esd.esd_xl_data_preprocessor.ESDXLDataPreprocessor(non_blocking=False)[source]

Bases: mmengine.model.base_model.data_preprocessor.BaseDataPreprocessor

ESDXLDataPreprocessor.

Parameters:

non_blocking (Optional[bool]) –

forward(data, training=False)[source]

Preprocesses the data into the model input format.

After the data pre-processing of cast_data(), forward will stack the input tensor list to a batch tensor at the first dimension.

Args:

data (dict): Data returned by dataloader training (bool): Whether to enable training time augmentation.

Returns:

dict or list: Data in the same format as the model input.

Parameters:

  • data (dict) –

  • training (bool) –

Return type:

Union[dict, list]