Bonus: PyTorch dataset class
You need to be a Pro member to access the full content of this.
Write PyTorch dataset class
In PyTorch, the Dataset
class is a useful abstraction that allows you to work with a large amount of data. To create a class for your own data, you need to define a class that inherits from torch.utils.data.Dataset
.
Let’s write the one for amazon-picking-challenge-2016
dataset we created in Export dataset.
Here is the base structure:
If you're a Pro member, sign in 🔒️ to read the rest of the guide. Or check our plans if you don't have them yet.