From e8cab09f1638389a6e95004c7a93604686cd19cf Mon Sep 17 00:00:00 2001
From: Oceania2018 <haiping008@gmail.com>
Date: Sat, 15 Feb 2020 12:46:49 -0600
Subject: [PATCH] global_step int64 #499

---
 src/TensorFlowNET.Core/Training/Optimizer.cs | 4 +++-
 src/TensorFlowNET.Hub/Tensorflow.Hub.csproj  | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/TensorFlowNET.Core/Training/Optimizer.cs b/src/TensorFlowNET.Core/Training/Optimizer.cs
index 524a0e34..04ec949c 100644
--- a/src/TensorFlowNET.Core/Training/Optimizer.cs
+++ b/src/TensorFlowNET.Core/Training/Optimizer.cs
@@ -205,7 +205,9 @@ namespace Tensorflow
                         //}
                         //else
                         {
-                            apply_updates = state_ops.assign_add(global_step, tf.constant(1), name: name);
+                            apply_updates = state_ops.assign_add(global_step,
+                                ops.convert_to_tensor(1, dtype: global_step.dtype),
+                                name: name);
                         }
                     });
                 }
diff --git a/src/TensorFlowNET.Hub/Tensorflow.Hub.csproj b/src/TensorFlowNET.Hub/Tensorflow.Hub.csproj
index 76965c1e..64ca76f6 100644
--- a/src/TensorFlowNET.Hub/Tensorflow.Hub.csproj
+++ b/src/TensorFlowNET.Hub/Tensorflow.Hub.csproj
@@ -2,7 +2,7 @@
   <PropertyGroup>
     <RootNamespace>Tensorflow.Hub</RootNamespace>
     <TargetFramework>netstandard2.0</TargetFramework>
-    <Version>0.0.6</Version>
+    <Version>0.1.0</Version>
     <Authors>Kerry Jiang, Haiping Chen</Authors>
     <Company>SciSharp STACK</Company>
     <Copyright>Apache 2.0</Copyright>