flash.core.data.transforms.kornia_collate¶

flash.core.data.transforms.kornia_collate(samples)[source]¶

Kornia transforms add batch dimension which need to be removed.

This function removes that dimension and then applies torch.utils.data._utils.collate.default_collate.