python train.py error

[Aniark]

$ python train.py
Traceback (most recent call last):
  File "/home/runner/work/debug/debug/FLAD/train.py", line 15, in <module>
    train_loader = DataLoader(train_ds, batch_size=batch, shuffle=True,
  File "/home/runner/.local/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 350, in __init__
    sampler = RandomSampler(dataset, generator=generator)  # type: ignore[arg-type]
  File "/home/runner/.local/lib/python3.10/site-packages/torch/utils/data/sampler.py", line 143, in __init__
    raise ValueError(f"num_samples should be a positive integer value, but got num_samples={self.num_samples}")
ValueError: num_samples should be a positive integer value, but got num_samples=0