deal with Go defer bug

golang/go#19710
unixpickle · Mar 25, 2017 · ec6ed89 · ec6ed89
1 parent e555500
commit ec6ed89
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/anyrnn/feedback.go b/anyrnn/feedback.go
@@ -167,7 +167,9 @@ func (f *feedbackRes) Propagate(u anyvec.Vector, s StateGrad,
 	g anydiff.Grad) (anyvec.Vector, StateGrad) {
 	for _, p := range []*anydiff.Var{f.InPool, f.LastOutPool} {
 		g[p] = p.Vector.Creator().MakeVector(p.Vector.Len())
-		defer delete(g, p)
+		defer func(g anydiff.Grad, p *anydiff.Var) {
+			delete(g, p)
+		}(g, p)
 	}
 	var blockUpstream StateGrad
 	if s != nil {