Fix model reloading bug

zhongkaifu · Sep 22, 2023 · 1ce1784 · 1ce1784
1 parent 8f2af2d
commit 1ce1784
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Seq2SeqSharp/Tools/WeightTensor.cs b/Seq2SeqSharp/Tools/WeightTensor.cs
@@ -499,11 +499,11 @@ public void SetWeightArray(float[] v)
                     halves[i] = new half(v[i]);
                 }
 
-                TWeight.SetElementsAsHalf(halves);
+                m_TWeight.SetElementsAsHalf(halves);
             }
             else
             {
-                TWeight.SetElementsAsFloat(v);
+                m_TWeight.SetElementsAsFloat(v);
             }
         }
 
@@ -514,7 +514,7 @@ public void SetWeightArray(half[] v)
                 throw new InvalidCastException($"Inconsistent element type in weights '{Name}'");
             }
 
-            TWeight.SetElementsAsHalf(v);
+            m_TWeight.SetElementsAsHalf(v);
         }
 
         public WeightTensor CopyWeightsRef(string name, bool needGradient, IComputeGraph graphToBind)