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

[satish-space]

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]

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));