Examples
============

TrainUnit Example
~~~~~~~~~~~~~~~~~~~~~
The TrainUnit example shows how to use the :class:`~torchtnt.framework.unit.TrainUnit` to train a basic model.

It can be found here: https://github.com/pytorch/tnt/blob/master/examples/train_unit_example.py


AutoUnit Example
~~~~~~~~~~~~~~~~~~~~~
The AutoUnit example shows how to use the :class:`~torchtnt.framework.auto_unit.AutoUnit` to train a basic model with
less code, and more training features enabled out of the box.

It can be found here: https://github.com/pytorch/tnt/blob/master/examples/auto_unit_example.py


TorchData Train Example
~~~~~~~~~~~~~~~~~~~~~~~~~
The TorchData Train example shows how to use TorchData's `DataPipe <https://pytorch.org/data/main/torchdata.datapipes.iter.html>`_ and `Dataloader2 <https://pytorch.org/data/main/dataloader2.html>`_. when training a basic model with TNT.

It can be found here: https://github.com/pytorch/tnt/blob/master/examples/torchdata_train_example.py


MNIST Example
~~~~~~~~~~~~~~~~~~~~~
The MNIST example shows how to use TNT to train a convnet model on the MNIST dataset.

It can be found here: https://github.com/pytorch/tnt/blob/master/examples/mnist/main.py


MinGPT Example
~~~~~~~~~~~~~~~~~~~~~
The MinGPT example shows how to use the :class:`~torchtnt.framework.auto_unit.AutoUnit` to train a MinGPT model.

It can be found here: https://github.com/pytorch/tnt/blob/master/examples/mingpt/main.py


TorchRec Example
~~~~~~~~~~~~~~~~~~~~~
The TorchRec example shows how to use :class:`~torchtnt.framework.auto_unit.TrainUnit` with `TorchRec <https://github.com/pytorch/torchrec>`_.

It can be found here: https://github.com/pytorch/tnt/blob/master/examples/torchrec/main.py