Transform¶
- class torchtune.modules.transforms.Transform(*args, **kwargs)[source]¶
Loose interface for all data and model transforms. Transforms operate at the sample level and perform operations on a sample dict, returning the updated dict. For an example implementation of this protocol, see
VisionCrossAttentionMask
.