|
select_idx = torch.multinomial(p_accept, 1, replacement=True).view(-1) |
@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!
M2m/train.py
Line 167 in 42d08a5
@jh-jeong
In the line above
p_acceptis supposed to be "per-class" probability distribution. This is according to the definition oftorch.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!