From c172126760d8a70833a44b404f8663c92f420b62 Mon Sep 17 00:00:00 2001 From: Samuel Caldas Date: Thu, 21 May 2020 22:58:38 -0300 Subject: [PATCH] Exposing epsilon in AdamOptimizer --- src/TensorFlowNET.Core/APIs/tf.train.cs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/TensorFlowNET.Core/APIs/tf.train.cs b/src/TensorFlowNET.Core/APIs/tf.train.cs index ca0ecc32..b3819b7b 100644 --- a/src/TensorFlowNET.Core/APIs/tf.train.cs +++ b/src/TensorFlowNET.Core/APIs/tf.train.cs @@ -38,8 +38,8 @@ namespace Tensorflow 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); + public Optimizer AdamOptimizer(float learning_rate, float epsilon = 1e-8f, string name = "Adam") + => new AdamOptimizer(learning_rate, epsilon:epsilon, name: name); public Optimizer AdamOptimizer(float learning_rate, TF_DataType dtype, string name = "Adam") => new AdamOptimizer(learning_rate, name: name, dtype: dtype);