main.py failed

[ggghamd]

https://github.com/yunjey/pytorch-tutorial/blob/master/tutorials/01-basics/pytorch_basics/main.py:

root@sriov-guest:~/dev-learn/gpu/pytorch/tutorial# python3 pytorch-main.py
tensor(2.)
tensor(1.)
tensor(1.)
w:  Parameter containing:
tensor([[-0.4404, -0.1190,  0.4829],
        [ 0.4956, -0.3760,  0.1435]], requires_grad=True)
b:  Parameter containing:
tensor([0.4760, 0.5684], requires_grad=True)
loss:  1.0722367763519287
dL/dw:  tensor([[-0.4775,  0.0949,  0.0065],
        [ 0.0698, -0.4295, -0.0664]])
dL/db:  tensor([0.3632, 0.3266])
loss after 1 step optimization:  1.0655767917633057
Downloading https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz to ../../data/cifar-10-python.tar.gz
170499072it [00:32, 5230791.96it/s]
Extracting ../../data/cifar-10-python.tar.gz to ../../data/
torch.Size([3, 32, 32])
6
Traceback (most recent call last):
  File "pytorch-main.py", line 157, in <module>
    shuffle=True)
  File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/dataloader.py", line 270, in __init__
    sampler = RandomSampler(dataset, generator=generator)  # type: ignore[arg-type]
  File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/sampler.py", line 103, in __init__
    "value, but got num_samples={}".format(self.num_samples))
ValueError: num_samples should be a positive integer value, but got num_samples=0
root@sriov-guest:~/dev-learn/gpu/pytorch/tutorial#

[rollysys]

The main.py is not a complete example code, it's some code snippets.
You shoud implement your own "custom data loader", refer to torchvision.datasets.CIFAR10.
Your error is caused by null implementation of CustomDataset.