Update GradientDescentOptimizerTests.cs

SciSharp · Jun 14, 2024 · 18db147 · 18db147
1 parent b3ce158
commit 18db147
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/TensorFlowNET.UnitTest/Training/GradientDescentOptimizerTests.cs b/test/TensorFlowNET.UnitTest/Training/GradientDescentOptimizerTests.cs
@@ -1,4 +1,4 @@
-using Microsoft.VisualStudio.TestTools.UnitTesting;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
 using System;
 using System.Linq;
 using Tensorflow;
@@ -82,7 +82,7 @@ private void TestMinimizeResourceVariable<T>() where T : struct
 
                 var pred = math_ops.matmul(var0, x) + var1;
                 var loss = pred * pred;
-                var sgd_op = tf.train.GradientDescentOptimizer(3.0f).minimize(loss);
+                var sgd_op = tf.train.GradientDescentOptimizer(1.0f).minimize(loss);
 
                 var global_variables = tf.global_variables_initializer();
                 sess.run(global_variables);