Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/torch/optim.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'doc/algos.md')
-rw-r--r--doc/algos.md3
1 files changed, 3 insertions, 0 deletions
diff --git a/doc/algos.md b/doc/algos.md
index a8dba9f..a3013d9 100644
--- a/doc/algos.md
+++ b/doc/algos.md
@@ -180,6 +180,8 @@ Arguments:
* `x`: the initial point
* `config`: a table with configuration parameters for the optimizer
* `config.learningRate`: learning rate
+ * `config.learningRateDecay`: learning rate decay
+ * `config.weightDecay`: weight decay coefficient for regularization
* `state`: a table describing the state of the optimizer; after each call the state is modified
* `state.paramVariance`: vector of temporal variances of parameters
@@ -201,6 +203,7 @@ Arguments:
* `config`: a table with configuration parameters for the optimizer
* `config.learningRate`: learning rate
* `config.learningRateDecay`: learning rate decay
+ * `config.weightDecay`: weight decay coefficient for regularization
* `config.beta1`: first moment coefficient
* `config.beta2`: second moment coefficient
* `config.epsilon`: for numerical stability