Hello, I am using self-built data set for model training, I encountered this problem during training, do you have a solution? Hope to get your reply. thank you。
File "train.py", line 204, in
main()
File "train.py", line 176, in main
merge_all_iters_to_one_epoch=args.merge_all_iters_to_one_epoch
File "/home/s/Desktop/LoGoNet-main/utils/al3d_utils/train_utils.py", line 93, in train_model
dataloader_iter=dataloader_iter
File "/home/s/Desktop/LoGoNet-main/utils/al3d_utils/train_utils.py", line 19, in train_one_epoch
batch = next(dataloader_iter)
File "/home/s/anaconda3/envs/logonet/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 521, in next
data = self._next_data()
File "/home/s/anaconda3/envs/logonet/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 1203, in _next_data
return self._process_data(data)
File "/home/s/anaconda3/envs/logonet/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 1229, in _process_data
data.reraise()
File "/home/s/anaconda3/envs/logonet/lib/python3.6/site-packages/torch/_utils.py", line 434, in reraise
raise exception
RuntimeError: Caught RuntimeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/home/s/anaconda3/envs/logonet/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop
data = fetcher.fetch(index)
File "/home/s/anaconda3/envs/logonet/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/s/anaconda3/envs/logonet/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 49, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/s/Desktop/LoGoNet-main/detection/al3d_det/datasets/kitti/kitti_dataset.py", line 477, in getitem
data_dict = self.prepare_data(data_dict=input_dict)
File "/home/s/Desktop/LoGoNet-main/detection/al3d_det/datasets/dataset_kitti.py", line 160, in prepare_data
data_dict=data_dict
File "/home/s/Desktop/LoGoNet-main/detection/al3d_det/datasets/processor/data_processor.py", line 197, in forward
data_dict = cur_processor(data_dict=data_dict)
File "/home/s/Desktop/LoGoNet-main/detection/al3d_det/datasets/processor/data_processor.py", line 102, in transform_points_to_voxels
voxel_output = self.voxel_generator.point_to_voxel(tv.from_numpy(points))
RuntimeError: /io/build/temp.linux-x86_64-3.6/spconv/build/src/csrc/sparse/all/ops_cpu3d/Point2VoxelCPU/Point2VoxelCPU_point_to_voxel_static.cc(22)
num_features == voxels.dim(2) assert faild. your points num features doesn't equal to voxel.
Hello, I am using self-built data set for model training, I encountered this problem during training, do you have a solution? Hope to get your reply. thank you。 File "train.py", line 204, in main() File "train.py", line 176, in main merge_all_iters_to_one_epoch=args.merge_all_iters_to_one_epoch File "/home/s/Desktop/LoGoNet-main/utils/al3d_utils/train_utils.py", line 93, in train_model dataloader_iter=dataloader_iter File "/home/s/Desktop/LoGoNet-main/utils/al3d_utils/train_utils.py", line 19, in train_one_epoch batch = next(dataloader_iter) File "/home/s/anaconda3/envs/logonet/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 521, in next data = self._next_data() File "/home/s/anaconda3/envs/logonet/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 1203, in _next_data return self._process_data(data) File "/home/s/anaconda3/envs/logonet/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 1229, in _process_data data.reraise() File "/home/s/anaconda3/envs/logonet/lib/python3.6/site-packages/torch/_utils.py", line 434, in reraise raise exception RuntimeError: Caught RuntimeError in DataLoader worker process 0. Original Traceback (most recent call last): File "/home/s/anaconda3/envs/logonet/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop data = fetcher.fetch(index) File "/home/s/anaconda3/envs/logonet/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/s/anaconda3/envs/logonet/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 49, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/s/Desktop/LoGoNet-main/detection/al3d_det/datasets/kitti/kitti_dataset.py", line 477, in getitem data_dict = self.prepare_data(data_dict=input_dict) File "/home/s/Desktop/LoGoNet-main/detection/al3d_det/datasets/dataset_kitti.py", line 160, in prepare_data data_dict=data_dict File "/home/s/Desktop/LoGoNet-main/detection/al3d_det/datasets/processor/data_processor.py", line 197, in forward data_dict = cur_processor(data_dict=data_dict) File "/home/s/Desktop/LoGoNet-main/detection/al3d_det/datasets/processor/data_processor.py", line 102, in transform_points_to_voxels voxel_output = self.voxel_generator.point_to_voxel(tv.from_numpy(points)) RuntimeError: /io/build/temp.linux-x86_64-3.6/spconv/build/src/csrc/sparse/all/ops_cpu3d/Point2VoxelCPU/Point2VoxelCPU_point_to_voxel_static.cc(22) num_features == voxels.dim(2) assert faild. your points num features doesn't equal to voxel.