Created by: Meiyim
fix #466 (closed)
softmax_with_cross_entropy
API 指定ignore_index
后依旧会发生越界;导致illegal memory access.
这个PR 通过显示指定loss_weights
而非ignore_index
来避免padding位置参与loss计算;从而规避了这个问题;
Created by: Meiyim
fix #466 (closed)
softmax_with_cross_entropy
API 指定ignore_index
后依旧会发生越界;导致illegal memory access.
这个PR 通过显示指定loss_weights
而非ignore_index
来避免padding位置参与loss计算;从而规避了这个问题;