Browse Source

assign_moving_average

tags/v0.12
Oceania2018 6 years ago
parent
commit
50d8d20b8e
2 changed files with 4 additions and 1 deletions
  1. +3
    -0
      src/TensorFlowNET.Core/APIs/tf.train.cs
  2. +1
    -1
      src/TensorFlowNET.Core/Train/moving_averages.cs

+ 3
- 0
src/TensorFlowNET.Core/APIs/tf.train.cs View File

@@ -31,6 +31,9 @@ namespace Tensorflow
public Optimizer AdamOptimizer(float learning_rate, string name = "Adam")
=> new AdamOptimizer(learning_rate, name: name);

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

public ExponentialMovingAverage ExponentialMovingAverage(float decay)
=> new ExponentialMovingAverage(decay);



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

@@ -25,7 +25,7 @@ namespace Tensorflow.Train
if (decay.dtype != variable.dtype.as_base_dtype())
decay = math_ops.cast(decay, variable.dtype.as_base_dtype());

return decay;
return state_ops.assign_sub(variable, (variable - value) * decay, name: scope);
});
}
}


Loading…
Cancel
Save