[BUG 记录] torch.multinomial/ CUDA error: device-side assert triggered

[BUG 记录] torch.multinomial/ CUDA error: device-side assert triggered

使用torch.multinomial时需要保证可取数量大于采样数,否则有些概率趋近于0的项永远不会被采集到。

解决办法:
1.设置replacement为True
2.对概率做个限制(clamp),防止某些决策的概率趋近于0

你可能感兴趣的:(人工智能,编程,神经网络,pytorch)