Browse Source

global_step int64 #499

tags/v0.20
Oceania2018 5 years ago
parent
commit
e8cab09f16
2 changed files with 4 additions and 2 deletions
  1. +3
    -1
      src/TensorFlowNET.Core/Training/Optimizer.cs
  2. +1
    -1
      src/TensorFlowNET.Hub/Tensorflow.Hub.csproj

+ 3
- 1
src/TensorFlowNET.Core/Training/Optimizer.cs View File

@@ -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);
}
});
}


+ 1
- 1
src/TensorFlowNET.Hub/Tensorflow.Hub.csproj View File

@@ -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>


Loading…
Cancel
Save