Browse Source

all tensor learning_rate for tf.train.GradientDescentOptimizer

tags/v0.12
Oceania2018 6 years ago
parent
commit
c4c6dbe2e5
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      src/TensorFlowNET.Core/APIs/tf.train.cs

+ 3
- 0
src/TensorFlowNET.Core/APIs/tf.train.cs View File

@@ -34,6 +34,9 @@ namespace Tensorflow
public Optimizer GradientDescentOptimizer(float learning_rate)
=> new GradientDescentOptimizer(learning_rate);

public Optimizer GradientDescentOptimizer(Tensor learning_rate)
=> new GradientDescentOptimizer(learning_rate);

public Optimizer AdamOptimizer(float learning_rate, string name = "Adam")
=> new AdamOptimizer(learning_rate, name: name);



Loading…
Cancel
Save