From 41cb39e35bc6f6a4c9d1e3b75d6764eb47b1c50c Mon Sep 17 00:00:00 2001 From: Oceania2018 Date: Wed, 16 Oct 2019 11:07:47 -0500 Subject: [PATCH] add Tesnor overload to GradientDescentOptimizer #422 --- src/TensorFlowNET.Core/TensorFlowNET.Core.csproj | 9 +++++---- src/TensorFlowNET.Core/Train/GradientDescentOptimizer.cs | 6 ++++++ 2 files changed, 11 insertions(+), 4 deletions(-) diff --git a/src/TensorFlowNET.Core/TensorFlowNET.Core.csproj b/src/TensorFlowNET.Core/TensorFlowNET.Core.csproj index 65645111..0f82d550 100644 --- a/src/TensorFlowNET.Core/TensorFlowNET.Core.csproj +++ b/src/TensorFlowNET.Core/TensorFlowNET.Core.csproj @@ -5,7 +5,7 @@ TensorFlow.NET Tensorflow 1.14.0 - 0.11.6 + 0.11.7 Haiping Chen, Meinrad Recheis, Eli Belash SciSharp STACK true @@ -17,7 +17,7 @@ TensorFlow, NumSharp, SciSharp, MachineLearning, TensorFlow.NET, C# Google's TensorFlow full binding in .NET Standard. Docs: https://tensorflownet.readthedocs.io - 0.11.6.0 + 0.11.7.0 Changes since v0.10.0: 1. Upgrade NumSharp to v0.20.3. 2. Add DisposableObject class to manage object lifetime. @@ -31,9 +31,10 @@ Docs: https://tensorflownet.readthedocs.io 10. Support n-dim indexing for tensor. 11. Add RegisterNoGradients 12. Add CumsumGrad, BroadcastToGrad. -13. Return VariableV1 instead of RefVariable. +13. Return VariableV1 instead of RefVariable. +14. 7.3 - 0.11.6.0 + 0.11.7.0 LICENSE true true diff --git a/src/TensorFlowNET.Core/Train/GradientDescentOptimizer.cs b/src/TensorFlowNET.Core/Train/GradientDescentOptimizer.cs index 2d472f5a..1a2821bb 100644 --- a/src/TensorFlowNET.Core/Train/GradientDescentOptimizer.cs +++ b/src/TensorFlowNET.Core/Train/GradientDescentOptimizer.cs @@ -39,6 +39,12 @@ namespace Tensorflow.Train : base(learning_rate, use_locking, name) { _lr = learning_rate; + } + + public GradientDescentOptimizer(Tensor learning_rate, bool use_locking = false, string name = "GradientDescent") + : base(learning_rate, use_locking, name) + { + _lr_t = learning_rate; } public override void _prepare()