KorniaParallelTransforms

class flash.core.data.transforms.KorniaParallelTransforms(*args: Any, **kwargs: Any)

The KorniaParallelTransforms class is an nn.Sequential which will apply the given transforms to each input (to .forward) in parallel, whilst sharing the random state (._params). This should be used when multiple elements need to be augmented in the same way (e.g. an image and corresponding segmentation mask).

Parameters

args – The transforms, passed to the nn.Sequential super constructor.