Getting Python Scalars Error while training the v2vnet

[xavierallem]

Hello, I have been trying to train the v2vnet from the checkpoint provided
-> Downloaded the model and loaded the Checkpoint Folder
-> changed the lr: 0.00001 , and Batch size to 2 from the hypes_yaml for v2vnet rest changed nothing.
-> Getting this error

TypeError: Caught TypeError in DataLoader worker process 0.
Original Traceback (most recent call last):
  File "/home/anaconda3/envs/opencood/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 302, in _worker_loop
    data = fetcher.fetch(index)
  File "/home/anaconda3/envs/opencood/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/home/anaconda3/envs/opencood/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 49, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/home/OpenCOOD/opencood/data_utils/datasets/intermediate_fusion_dataset.py", line 58, in __getitem__
    cur_ego_pose_flag=self.cur_ego_pose_flag)
  File "/home/OpenCOOD/opencood/data_utils/datasets/basedataset.py", line 236, in retrieve_base_data
    self.calc_dist_to_ego(scenario_database, timestamp_key)
  File "/home/OpenCOOD/opencood/data_utils/datasets/basedataset.py", line 339, in calc_dist_to_ego
    (cur_lidar_pose[1] - ego_lidar_pose[1]) ** 2)
TypeError: only size-1 arrays can be converted to Python scalars

What can be the issue here?