diff options
Diffstat (limited to 'src/training/graph_group_async.cpp')
-rw-r--r-- | src/training/graph_group_async.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/training/graph_group_async.cpp b/src/training/graph_group_async.cpp index 72b06e48..f85f9cf8 100644 --- a/src/training/graph_group_async.cpp +++ b/src/training/graph_group_async.cpp @@ -143,13 +143,13 @@ void AsyncGraphGroup::execute(Ptr<data::Batch> batch) { thread_local Tensor accGradients; thread_local Ptr<TensorAllocator> accAlloc; - ABORT_IF(costScale_ ,"Cost-scaling not implemented for AsyncSGD"); + ABORT_IF(costScaling_ ,"Cost-scaling not implemented for AsyncSGD"); auto graph = graphs_[tid]; Ptr<RationalLoss> dynamicLoss = models_[tid]->build(graph, batch); - if(costScaleFactor_ != 1.f) { + if(costScalingFactor_ != 1.f) { // it's ok to go out of scope, this will still insert the new top node into the graph - auto costNode = dynamicLoss->loss() * costScaleFactor_; + auto costNode = dynamicLoss->loss() * costScalingFactor_; } if(t % optimizerDelay_ == 0) { |