-
Notifications
You must be signed in to change notification settings - Fork 77
Description
I was running the example of 3DFMNet, then I encountered this error message:
Traceback (most recent call last):
File "lib/robiFocus_test.py", line 225, in
main()
File "lib/robiFocus_test.py", line 214, in main
run_one_epoch(
File "lib/robiFocus_test.py", line 132, in run_one_epoch
for i, data_dict in enumerate(data_loader):
File "/root/miniconda3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 630, in next
data = self._next_data()
File "/root/miniconda3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1344, in _next_data
return self._process_data(data)
File "/root/miniconda3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1370, in _process_data
data.reraise()
File "/root/miniconda3/lib/python3.8/site-packages/torch/_utils.py", line 706, in reraise
raise exception
TypeError: Caught TypeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/root/miniconda3/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 309, in _worker_loop
data = fetcher.fetch(index) # type: ignore[possibly-undefined]
File "/root/miniconda3/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 55, in fetch
return self.collate_fn(data)
File "/root/autodl-tmp/3DFMNet-main/vision3d/datasets/registration/threedmatch_kpconv_v1.py", line 1046, in threedmatch_kpconv_collate_fn
input_points, input_neighbors, input_pools, input_upsamples, input_lengths = generate_input_data_gpu(
File "/root/autodl-tmp/3DFMNet-main/vision3d/modules/kpconv/helpers.py", line 250, in generate_input_data_gpu
conv_i = batch_neighbors_kpconv_gpu(
File "/root/miniconda3/lib/python3.8/site-packages/torch/utils/_contextlib.py", line 116, in decorate_context
return func(*args, **kwargs)
File "/root/autodl-tmp/3DFMNet-main/vision3d/modules/kpconv/helpers.py", line 35, in batch_neighbors_kpconv_gpu
knn_distances, knn_indices = knn_func(cur_q_points, cur_s_points)
File "/root/miniconda3/lib/python3.8/site-packages/pykeops/common/lazy_tensor.py", line 957, in call
return self.callfun(*args, *self.variables, **self.kwargs)
File "/root/miniconda3/lib/python3.8/site-packages/pykeops/torch/generic/generic_red.py", line 693, in call
out = GenredAutograd_fun(params, *args)
File "/root/miniconda3/lib/python3.8/site-packages/pykeops/torch/generic/generic_red.py", line 383, in GenredAutograd_fun
return GenredAutograd.apply(*inputs)[0]
File "/root/miniconda3/lib/python3.8/site-packages/torch/autograd/function.py", line 574, in apply
return super().apply(*args, **kwargs) # type: ignore[misc]
File "/root/miniconda3/lib/python3.8/site-packages/pykeops/torch/generic/generic_red.py", line 291, in forward
return GenredAutograd_base._forward(*inputs)
File "/root/miniconda3/lib/python3.8/site-packages/pykeops/torch/generic/generic_red.py", line 91, in _forward
myconv = keops_binder["nvrtc" if tagCPUGPU else "cpp"](
File "/root/miniconda3/lib/python3.8/site-packages/keopscore/utils/Cache.py", line 92, in call
obj = self.cls(*args)
File "/root/miniconda3/lib/python3.8/site-packages/pykeops/common/keops_io/LoadKeOps_nvrtc.py", line 16, in init
super().init(*args, fast_init=fast_init)
File "/root/miniconda3/lib/python3.8/site-packages/pykeops/common/keops_io/LoadKeOps.py", line 18, in init
self.init(*args)
File "/root/miniconda3/lib/python3.8/site-packages/pykeops/common/keops_io/LoadKeOps.py", line 127, in init
) = get_keops_dll(
File "/root/miniconda3/lib/python3.8/site-packages/keopscore/utils/Cache.py", line 33, in call
self.library[str_id] = self.fun(*args)
File "/root/miniconda3/lib/python3.8/site-packages/keopscore/get_keops_dll.py", line 97, in get_keops_dll_impl
red_formula = GetReduction(red_formula_string, aliases)
File "/root/miniconda3/lib/python3.8/site-packages/keopscore/formulas/GetReduction.py", line 33, in new
reduction = eval(red_formula_string, globals(), aliases_dict)
File "", line 1, in
TypeError: init() missing 1 required positional argument: 'tagIJ'
May I ask how to solve this problem? Is it a version issue?
The versions of the relevant libraries are as follows:
torch 2.4.0+cu118
torch-cluster 1.6.3+pt24cu118
torch-scatter 2.1.2+pt24cu118
vision3d 0.0.1
pykeops 2.3
keopscore 2.3