cerebras.modelzoo.data.vision.transforms.create_transform#

cerebras.modelzoo.data.vision.transforms.create_transform(transform_spec)[source]#

Create the specified transform. For each transform, the parameter list (name and default value) follows those in torchvision 0.12 (https://pytorch.org/vision/0.12/transforms.html)

Parameters

  • name (str) – name of the transform

  • args (dict) – a dictionary of parameters used to initialize the transform. Default is None.

previous

cerebras.modelzoo.data.vision.transforms

next

cerebras.modelzoo.data.vision.utils