pytorch multinomial 报错 device-side assert triggered

/pytorch/aten/src/THC/THCTensorRandom.cuh:187: void sampleMultinomialOnce(long *, long, int, T *, T *, int, int) [with T = float, AccT = float]: block: [6,0,0], thread: [5,0,0] Assertion `THCNumerics::ge(val, zero)` failed.

因为multinomial的输入

must be non-negative, finite and have a non-zero sum

具体见doc

可能是上游出现nan了

nan可能是loss引起的训练不稳定导致,

你可能感兴趣的:(PyTorch)