IndexError: index 3678 is out of bounds for dimension 1 with size 3024

[raw5]

I keep getting IndexError: index 3678 is out of bounds for dimension 1 with size 3024 when training my models. May I know what I am doing wrong?

Hello Wooden
Load data: Begin
Load data: End
0%| | 0/300000 [00:00<?, ?it/s]
Traceback (most recent call last):
File "exp_runner.py", line 392, in
runner.train()
File "exp_runner.py", line 105, in train
data = self.dataset.gen_random_rays_at(image_perm[self.iter_step % len(image_perm)], self.batch_size)
File "/home/user/NeuS/models/dataset.py", line 119, in gen_random_rays_at
mask = self.masks[img_idx][(pixels_y, pixels_x)] # batch_size, 3
IndexError: index 3678 is out of bounds for dimension 1 with size 3024

[newforests]

May I ask if you have solved this problem? Is there any corresponding solution recommended？