/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<T>::ge(val, zero)` failed.
因为multinomial的输入
must be non-negative, finite and have a non-zero sum
具体见doc
可能是上游出现nan了
nan可能是loss引起的训练不稳定导致,