Code runs successfully on CPU!
Yes, I have checked the index values, they are very extreme 4
I tried to print the tensors and found that values of tensor inds_g is too large and too small(which casuses out of bounds error)
inds_g.min() = tensor(-4993021444723710459)
inds_g.max() = tensor(4575432887736600530)
inds_g = tensor([[[ 4255818524050935954, 62],
[ 4256250760978027750, 62],
[ 4237722774569238629, 62],
…,]]
usage of this tensor:
file “run_nerf_helpers.py”
cdf_g = torch.gather(cdf.unsqueeze(1).expand(matched_shape), 2, inds_g)