From c93c21994c9a9e90f3d2164b4892494005736efb Mon Sep 17 00:00:00 2001
From: Oceania2018 <haiping008@gmail.com>
Date: Sun, 28 Nov 2021 16:34:30 -0600
Subject: [PATCH] GC.Collect in ComputeGradient

---
 src/TensorFlowNET.Core/Gradients/Tape.ComputeGradient.cs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/TensorFlowNET.Core/Gradients/Tape.ComputeGradient.cs b/src/TensorFlowNET.Core/Gradients/Tape.ComputeGradient.cs
index 0d0ecbe2..3a5f4efa 100644
--- a/src/TensorFlowNET.Core/Gradients/Tape.ComputeGradient.cs
+++ b/src/TensorFlowNET.Core/Gradients/Tape.ComputeGradient.cs
@@ -26,6 +26,7 @@ namespace Tensorflow.Gradients
                 tensor_tape_,
                 state.op_tape);
 
+            int gcCollectFlag = 0;
             while (!op_stack.empty())
             {
                 var op = op_stack.Dequeue();
@@ -154,6 +155,9 @@ namespace Tensorflow.Gradients
                             op_stack.Enqueue(op_id);
                     }
                 }
+
+                if (gcCollectFlag++ % 10 == 0)
+                    GC.Collect();
             }
 
             if (state.op_tape.Count > 0)