KorniaParallelTransforms¶
- class flash.core.data.transforms.KorniaParallelTransforms(*args: Any, **kwargs: Any)[source]¶
The
KorniaParallelTransforms
class is annn.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.