emmi.pipeline.collators.default¶

Classes¶

Applies torch.utils.data.default_collate to the specified items.

class emmi.pipeline.collators.default.DefaultCollator(items=None, optional_items=None)¶

Applies torch.utils.data.default_collate to the specified items.

Initializes the DefaultCollator.

Parameters:

items (list[str] | None) – Items to apply default_collate to.
optional_items (list[str] | None) – Items to apply default_collate to if they are present.