Browse Source

Add default name for AdamOptimizer.

tags/v0.9
Oceania2018 6 years ago
parent
commit
69f510190f
2 changed files with 2 additions and 3 deletions
  1. +1
    -1
      src/TensorFlowNET.Core/Train/tf.optimizers.cs
  2. +1
    -2
      test/TensorFlowNET.Examples/TextProcess/CnnTextClassification.cs

+ 1
- 1
src/TensorFlowNET.Core/Train/tf.optimizers.cs View File

@@ -13,7 +13,7 @@ namespace Tensorflow
public static Optimizer GradientDescentOptimizer(float learning_rate)
=> new GradientDescentOptimizer(learning_rate);

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

public static Saver Saver(VariableV1[] var_list = null) => new Saver(var_list: var_list);


+ 1
- 2
test/TensorFlowNET.Examples/TextProcess/CnnTextClassification.cs View File

@@ -22,7 +22,7 @@ namespace TensorFlowNET.Examples
public bool Enabled { get; set; } = true;
public string Name => "CNN Text Classification";
public int? DataLimit = null;
public bool IsImportingGraph { get; set; } = true;
public bool IsImportingGraph { get; set; } = false;

private const string dataDir = "word_cnn";
private string dataFileName = "dbpedia_csv.tar.gz";
@@ -304,7 +304,6 @@ namespace TensorFlowNET.Examples
public bool Train()
{
var graph = IsImportingGraph ? ImportGraph() : BuildGraph();
// string json = JsonConvert.SerializeObject(graph, Formatting.Indented);
return with(tf.Session(graph), sess => Train(sess, graph));
}



Loading…
Cancel
Save