提交 029be5fd 编写于 作者: L liuhongyu

fix lstmp bug; test=develop

上级 393fa602
......@@ -405,10 +405,10 @@ class LSTMPGradKernel : public framework::OpKernel<T> {
}
int cur_batch_size = bend - bstart;
// lstm_value.output_value not used in bp, set to null
// lstm_grad.state_active_grad not used in bp, set to null
lstm_value.output_value = nullptr;
lstm_grad.state_active_grad = nullptr;
// lstmp_value.output_value not used in bp, set to null
// lstmp_grad.state_active_grad not used in bp, set to null
lstmp_value.output_value = nullptr;
lstmp_grad.state_active_grad = nullptr;
math::LstmUnitGradFunctor<DeviceContext, T>::compute(
device_ctx, lstmp_value, lstmp_grad, frame_size, cur_batch_size,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册