Module collate

Merges a list of samples to form a batch.

Structs

DefaultCollate

Default Collate function that mimic the default_collate function from PyTorch.

NoOpCollate

Simple Collate that doesn’t change the batch of samples.

TorchCollatetch

Torch Collate function that mimic the default_collate function from PyTorch.

Traits

Collate

Any collate gather samples from one batch together.