nntrainer/optimizers/sgd.cpp

   1 // SPDX-License-Identifier: Apache-2.0
   2 /**
   3  * Copyright (C) 2020 Parichay Kapoor <pk.kapoor@samsung.com>
   4  *
   5  * @file   sgd.cpp
   6  * @date   6 October 2020
   7  * @see    https://github.com/nnstreamer/nntrainer
   8  * @author Jijoong Moon <jijoong.moon@samsung.com>
   9  * @author Parichay Kapoor <pk.kapoor@samsung.com>
  10  * @bug    No known bugs except for NYI items
  11  * @brief  This is the SGD optimizer.
  12  */
  13
  14 #include <sgd.h>
  15
  16 namespace nntrainer {
  17
  18 SGD::SGD() { setProperty({"learning_rate=0.0001"}); }
  19
  20 void SGD::applyGradient(RunOptimizerContext &context) {
  21   context.applyGradient(context.getLearningRate());
  22 }
  23
  24 } // namespace nntrainer