Add memset to initialize gradients in CTC wrapper, when gradients are allocated

上级 a95aa1ad
......@@ -213,6 +213,10 @@ int APPLY_SPECIFIC(ctc_cost_gpu)(PyGpuArrayObject * in_activations,
return 1;
}
}
else
{
GpuArray_memset( &((*out_gradients)->ga), 0 );
}
switch ( (*out_gradients)->ga.typecode )
{
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论