Shortcuts

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.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources