Traceback (most recent call last):
File "E:\project\test\DH_live\train\train_render_model.py", line 104, in
for iteration, data in enumerate(training_data_loader):
File "E:\project\test\DH_live\py311\Lib\site-packages\torch\utils\data\dataloader.py", line 630, in next
data = self._next_data()
^^^^^^^^^^^^^^^^^
File "E:\project\test\DH_live\py311\Lib\site-packages\torch\utils\data\dataloader.py", line 673, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\project\test\DH_live\py311\Lib\site-packages\torch\utils\data_utils\fetch.py", line 52, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "E:\project\test\DH_live\py311\Lib\site-packages\torch\utils\data_utils\fetch.py", line 52, in
data = [self.dataset[idx] for idx in possibly_batched_index]
~~~~~~~~~~~~^^^^^
File "E:\project\test\DH_live\talkingface\data\few_shot_dataset.py", line 151, in getitem
target_keypoints = self.driving_keypoints[video_index][current_clip]
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^
IndexError: index 150 is out of bounds for axis 0 with size 150

在训练时出现这个问题，但是当我测试第一轮的时候有些是正常的，现在大概是几十个视频，每个视频大概是5s