batch_size>1 get wrong

[RONINGOD]

File "~/miniconda3/envs/pointocc/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop data = fetcher.fetch(index) File "~/miniconda3/envs/pointocc/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 52, in fetch return self.collate_fn(data) File "~/project/PointOcc/dataloader/dataset_wrapper.py", line 303, in occ_custom_collate_fn grid_ind_stack = np.stack([d[2] for d in data]).astype(np.float32) File "<array_function internals>", line 200, in stack File "~/miniconda3/envs/pointocc/lib/python3.8/site-packages/numpy/core/shape_base.py", line 464, in stack raise ValueError('all input arrays must have the same shape') ValueError: all input arrays must have the same shape

[Naclzno]

Hello！I have the same problem. Do you have any solutions?