Skip to content

Latest commit

 

History

History
21 lines (16 loc) · 740 Bytes

pytorch.rst

File metadata and controls

21 lines (16 loc) · 740 Bytes
Raw

PyTorch

This topic describes how to integrate TensorBay dataset with PyTorch Pipeline using the MNIST Dataset as an example.

The typical method to integrate TensorBay dataset with PyTorch is to build a "Segment" class derived from torch.utils.data.Dataset.

../../../docs/code/use_dataset_in_pytorch.py

Using the following code to create a PyTorch dataloader and run it:

../../../docs/code/use_dataset_in_pytorch.py