Bug fix for GPU memory save mode.

zhongkaifu · Feb 12, 2024 · 3ab4c75 · 3ab4c75
1 parent b6def4b
commit 3ab4c75
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Seq2SeqSharp/Tools/ComputeGraphTensor.cs b/Seq2SeqSharp/Tools/ComputeGraphTensor.cs
@@ -102,7 +102,7 @@ public IWeightFactory GetWeightFactory()
 
         public IComputeGraph CreateSubGraph(string name)
         {
-            ComputeGraphTensor subGraph = new ComputeGraphTensor(m_weightTensorFactory, m_deviceId, m_needsBackprop, m_backprop, isSubGraph: true);
+            ComputeGraphTensor subGraph = new ComputeGraphTensor(m_weightTensorFactory, m_deviceId, m_needsBackprop, m_backprop, isSubGraph: true, saveGPUMemoryMode:m_saveGPUMemoryMode);
             return subGraph;
         }