Error when running this: "python setup.py install --user"

Duankaiwen / CenterNet

Codes for our paper "CenterNet: Keypoint Triplets for Object Detection" .

MIT License

1.86k stars 384 forks source link

Error when running this: "python setup.py install --user" #141

Closed satish-space closed 2 years ago

satish-space commented 2 years ago

Here is the error:

/usr/local/lib/python3.6/site-packages/torch/include/ATen/Functions.h:6078:29: note: candidate expects 5 arguments, 2 provided src/top_pool.cpp:45:78: error: no matching function for call to ‘zeros(at::DeprecatedTypeProperties&, )’ auto max_ind = at::zeros(torch::CUDA(at::kLong), {batch, channel, width});

satish-space commented 2 years ago

The issue is fixed by updating the order of input parameters.

Changed auto max_ind = at::zeros(torch::CUDA(at::kLong), {batch, channel, width}); By auto max_ind = at::zeros({batch, channel, width}, torch::CUDA(at::kLong));