Write a Custom Dataset and DataLoader in PyTorch
Basicauto-graded
Create a custom Dataset class and use DataLoader to batch, shuffle, and iterate over your data for training.
Solve it
Check your answer
The grader verifies properties of your implementation, so a correct solution written differently from ours still passes.
pip install torchleet
from torchleet import check
check("custom-dataset", LinearRegressionDataset)