Error message shown at train_audiovisual.sh

[chikiuso]

Hi, it shows the following errors msg when I go step 2 (train_audiovisual.sh), could you help? thanks!

Traceback (most recent call last):
File "train_audiovisual.py", line 77, in
G.optimize_parameters_video(dataloader, args.epochs, device, display_freq=args.display_freq, save_freq=args.save_freq, save_dir=save_dir, experimentName=experimentName, initial_niter=args.initial_iter, load_model=args.load_model)
File "/root/newfacegen/model_vc.py", line 658, in optimize_parameters_video
for i, data in enumerate(dataloader):
File "/usr/local/lib/python3.5/dist-packages/torch/utils/data/dataloader.py", line 582, in next
return self._process_next_batch(batch)
File "/usr/local/lib/python3.5/dist-packages/torch/utils/data/dataloader.py", line 608, in _process_next_batch
raise batch.exc_type(batch.exc_msg)
IndexError: Traceback (most recent call last):
File "/usr/local/lib/python3.5/dist-packages/torch/utils/data/_utils/worker.py", line 99, in _worker_loop
samples = collate_fn([dataset[i] for i in batch_indices])
File "/usr/local/lib/python3.5/dist-packages/torch/utils/data/_utils/worker.py", line 99, in
samples = collate_fn([dataset[i] for i in batch_indices])
File "/root/newfacegen/data/Sample_dataset.py", line 165, in getitem
video[j, :, :, :] = self.transform(Image.open(self.list_frame[j + video_index]).convert('RGB'))
IndexError: list index out of range