Get Started

Install TorchStudio, load the MNIST dataset and train a model to recognize handwritten digits.

Load Custom Datasets

Use the GenericLoader dataset to load classification and segmentation datasets and apply transforms.

Transfer Learning

Format the CIFAR10 dataset, train untrained and pre-trained models and see how their performance compare.

Fine-Tune a Model

Edit the CMUARCTIC dataset to fine-tune a customizable model on repairing clipped audio waveforms.

Use Remote Servers

Prepare, add and use a remote server for dataset loading, model training and inference.

IDE Extensions

Install the extensions, add TorchStudio to your IDE project, drag your scripts, and export the weights back to your IDE.

Videos