Hi,
Thank you for your great work.
in the lines 66, 67 of common.py module you have:
pdf = weights / torch.sum(weights, -1, keepdim=True)
pdf = weights
in the second line you override pdf = weights / torch.sum(weights, -1, keepdim=True)
so why having the first line at all?