Per-class probablibity being treated as per-sample probability.
souravraha opened this issue · comments
Line 167 in 42d08a5
@jh-jeong
In the line above
p_accept
is supposed to be "per-class" probability distribution. This is according to the definition of torch.multinomial
. However, the implementation is treating it as a per-sample probability distribution. This is not how the subsection "Practical implementation via re-sampling" of https://arxiv.org/pdf/2004.00431.pdf describes this to be implemented. Could you shed some light on this? Thanks!