[BUG 记录] torch.multinomial/ CUDA error: device-side assert triggered
使用torch.multinomial时需要保证可取数量大于采样数,否则有些概率趋近于0的项永远不会被采集到。
解决办法:
1.设置replacement为True
2.对概率做个限制(clamp),防止某些决策的概率趋近于0
[BUG 记录] torch.multinomial/ CUDA error: device-side assert triggered
使用torch.multinomial时需要保证可取数量大于采样数,否则有些概率趋近于0的项永远不会被采集到。
解决办法:
1.设置replacement为True
2.对概率做个限制(clamp),防止某些决策的概率趋近于0