IndexError: index 4 is out of bounds for axis 0 with size 3

[20LGWZQ]

IndexError: Caught IndexError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "D:\acaconda\envs\mmpose\lib\site-packages\torch\utils\data_utils\worker.py", line 202, in _worker_loop
data = fetcher.fetch(index)
File "D:\acaconda\envs\mmpose\lib\site-packages\torch\utils\data_utils\fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "D:\acaconda\envs\mmpose\lib\site-packages\torch\utils\data_utils\fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "C:\Users\lvelu\Desktop\OpenSitUp-main\OpenSitUp-main\Trainer\data\person_keypoint_txt.py", line 219, in getitem
heatmap_c = heatmap[labels_kp_name]
IndexError: index 4 is out of bounds for axis 0 with size 3

博主你好，我在用你的代码跑我自己的模型的时候，我自己的数据集关键点数量是6个，我改了cfg文件里的配置信息，但是在运行的时候在person_keypoint_txt.py问价卡住，报错如上所述，请问怎么解决呀，冒昧打扰