Real3DPortrait icon indicating copy to clipboard operation
Real3DPortrait copied to clipboard

StopIteration - Train audio_lm3d_syncnet [Training Error]

Open NithinJangid opened this issue 1 year ago • 0 comments

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

NithinJangid avatar Aug 14 '24 11:08 NithinJangid