XavierXiao / Dreambooth-Stable-Diffusion

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion
MIT License
7.57k stars 790 forks source link

ValueError: `Dataloader` returned 0 length. Please make sure that it returns at least 1 batch #120

Open jeremy-linchao opened 1 year ago

jeremy-linchao commented 1 year ago

when running the script

python main.py --base configs/stable-diffusion/v1-finetune_unfrozen.yaml -t --actual_resume /path/to/original/stable-diffusion/sd-v1-4-full-ema.ckpt
-n --gpus 0, --data_root /root/to/training/images --reg_data_root /root/to/regularization/images --class_word

it raise the exception

Traceback (most recent call last): File "main.py", line 830, in trainer.fit(model, data) File "/root/anaconda3/envs/ldm/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 553, in fit self._run(model) File "/root/anaconda3/envs/ldm/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 918, in _run self._dispatch() File "/root/anaconda3/envs/ldm/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 986, in _dispatch self.accelerator.start_training(self) File "/root/anaconda3/envs/ldm/lib/python3.8/site-packages/pytorch_lightning/accelerators/accelerator.py", line 92, in start_training self.training_type_plugin.start_training(trainer) File "/root/anaconda3/envs/ldm/lib/python3.8/site-packages/pytorch_lightning/plugins/training_type/training_type_plugin.py", line 161, in start_training self._results = trainer.run_stage() File "/root/anaconda3/envs/ldm/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 996, in run_stage return self._run_train() File "/root/anaconda3/envs/ldm/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 1031, in _run_train self._run_sanity_check(self.lightning_module) File "/root/anaconda3/envs/ldm/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 1111, in _run_sanity_check self._evaluation_loop.reload_evaluation_dataloaders() File "/root/anaconda3/envs/ldm/lib/python3.8/site-packages/pytorch_lightning/loops/dataloader/evaluation_loop.py", line 173, in reload_evaluation_dataloaders self.trainer.reset_val_dataloader(model) File "/root/anaconda3/envs/ldm/lib/python3.8/site-packages/pytorch_lightning/trainer/data_loading.py", line 437, in reset_val_dataloader self.num_val_batches, self.val_dataloaders = self._reset_eval_dataloader(model, "val") File "/root/anaconda3/envs/ldm/lib/python3.8/site-packages/pytorch_lightning/trainer/data_loading.py", line 398, in _reset_eval_dataloader num_batches = len(dataloader) if has_len(dataloader) else float("inf") File "/root/anaconda3/envs/ldm/lib/python3.8/site-packages/pytorch_lightning/utilities/data.py", line 63, in has_len raise ValueError("Dataloader returned 0 length. Please make sure that it returns at least 1 batch") ValueError: Dataloader returned 0 length. Please make sure that it returns at least 1 batch

could anyone help me

williamyorkl commented 1 year ago

same here bro~ did you solve the problem?