code: def swav_transform(
      total_num_crops: int = 8,
      num_crops: Sequence[int] = [2, 6],
      size_crops: Sequence[int] = [224, 96],
      crop_scales: Sequence[Sequence[float]] = [[0.4, 1], [0.05, 0.4]],
      gaussian_blur: bool = True,
      jitter_strength: float = 1.0,
      normalize: Optional[nn.Module] = None,
      collate_fn: Callable = multicrop_collate_fn,
  ) -> partial:
      """For swav."""
      return partial(
          StandardMultiCropSSLTransform,
          total_num_crops=total_num_crops,
          num_crops=num_crops,
          size_crops=size_crops,
          crop_scales=crop_scales,
          gaussian_blur=gaussian_blur,
          jitter_strength=jitter_strength,
          normalize=normalize,
          collate_fn=collate_fn,
      )
  
file_path: lightning-flash/src/flash/image/embedding/transforms/vissl_transforms.py
imports: from functools import partial
  from typing import Callable, Optional, Sequence
  import torch.nn
  from flash.core.registry import FlashRegistry
  from flash.image.embedding.vissl.transforms import multicrop_collate_fn, simclr_collate_fn
  from flash.image.embedding.vissl.transforms.multicrop import StandardMultiCropSSLTransform
label_desc: Use image node_id 2 for calling a global function with example usage: swav_transform(total_num_crops, num_crops, size_crops, crop_scales, gaussian_blur, jitter_strength, normalize, collate_fn) and returns: partial