Merge pull request #5010 from tensor-tang/merge

fix crash on ds2
revert-4814-Add_sequence_project_op
Tao Luo 7 years ago committed by GitHub
commit 48173e8599

@ -339,11 +339,15 @@ private:
* clear all grad
*/
void clearGrads() {
if (output_.grad) {
output_.grad->zeroMem();
}
for (size_t i = 0; i < outputOtherDevice_.size(); i++) {
if (outputOtherDevice_[i].grad) {
outputOtherDevice_[i].grad->zeroMem();
}
}
}
/**
* Set deviceId of the params used in this layer.

Loading…
Cancel
Save