Browse Source

Update GradientDescentOptimizerTests.cs

pull/1217/head
novikov-alexander GitHub 1 year ago
parent
commit
18db147eb4
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      test/TensorFlowNET.UnitTest/Training/GradientDescentOptimizerTests.cs

+ 2
- 2
test/TensorFlowNET.UnitTest/Training/GradientDescentOptimizerTests.cs View File

@@ -1,4 +1,4 @@
using Microsoft.VisualStudio.TestTools.UnitTesting;
using Microsoft.VisualStudio.TestTools.UnitTesting;
using System; using System;
using System.Linq; using System.Linq;
using Tensorflow; using Tensorflow;
@@ -82,7 +82,7 @@ namespace TensorFlowNET.UnitTest.Training


var pred = math_ops.matmul(var0, x) + var1; var pred = math_ops.matmul(var0, x) + var1;
var loss = pred * pred; 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(); var global_variables = tf.global_variables_initializer();
sess.run(global_variables); sess.run(global_variables);


Loading…
Cancel
Save