kohya_ss icon indicating copy to clipboard operation
kohya_ss copied to clipboard

ValueError: num_samples should be a positive integer value, but got num_samples=0

Open popovidis opened this issue 7 months ago • 3 comments

Traceback (most recent call last):
  File "/workspace/kohya_ss/sd-scripts/train_db.py", line 529, in <module>
    train(args)
  File "/workspace/kohya_ss/sd-scripts/train_db.py", line 190, in train
    train_dataloader = torch.utils.data.DataLoader(
  File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/dataloader.py", line 350, in __init__
    sampler = RandomSampler(dataset, generator=generator)  # type: ignore[arg-type]
  File "/usr/local/lib/python3.10/dist-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
Traceback (most recent call last):
  File "/workspace/kohya_ss/venv/bin/accelerate", line 8, in <module>
    sys.exit(main())
  File "/workspace/kohya_ss/venv/lib/python3.10/site-packages/accelerate/commands/accelerate_cli.py", line 47, in main
    args.func(args)
  File "/workspace/kohya_ss/venv/lib/python3.10/site-packages/accelerate/commands/launch.py", line 1017, in launch_command
    simple_launcher(args)
  File "/workspace/kohya_ss/venv/lib/python3.10/site-packages/accelerate/commands/launch.py", line 637, in simple_launcher
    raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd)
subprocess.CalledProcessError: Command '['/workspace/kohya_ss/venv/bin/python3', '/workspace/kohya_ss/sd-scripts/train_db.py', '--config_file', '/workspace/kohya_ss/outputs/model/config_dreambooth-20240721-213429.toml']' returned non-zero exit status 1.

Not sure what I am doing wrong, worked before...

This happens for both Lora and Model training.

popovidis avatar Jul 21 '24 20:07 popovidis