StopIteration - Train audio_lm3d_syncnet [Training Error]
When i ran the CMD: CUDA_VISIBLE_DEVICES=0 python tasks/run.py --config=egs/os_avatar/audio_lm3d_syncnet.yaml --exp_name=audio_lm3d_syncnet --reset.
i am getting StopIteration Error and when i checked the dataloader using len(dataloader) the length is 0, i dont know why
Console --->>>
| model Trainable Parameters: 10.746M | clip_loss_fn Trainable Parameters: 0.000M | hubert_encoder Trainable Parameters: 5.374M | mouth_encoder Trainable Parameters: 5.372M | logloss Trainable Parameters: 0.000M Sanity Val: 20%|██ | 1/5 [00:01<00:04, 1.13s/step] | Validation results@0: {'total_loss': 8000.6805884242, 'sync_loss': 0.6805884242, 'batch_size': 8000.0} 0step [00:00, ?step/s] Traceback (most recent call last): File "/media/xxx/Real3DPortrait/utils/commons/trainer.py", line 341, in train batch_idx, batch = next(train_iterator) StopIteration
During handling of the above exception, another exception occurred:
Traceback (most recent call last): File "/media/xxx/Real3DPortrait/utils/commons/trainer.py", line 236, in run_single_process self.train() File "/media/xxx/Real3DPortrait/utils/commons/trainer.py", line 344, in train batch_idx, batch = next(train_iterator) StopIteration