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