ほとんどのTensorflowコードで、Adam Optimizerが一定の学習率1e-4
(つまり0.0001)で使用されているのを見ました。通常、コードは次のようになります。
...build the model...
# Add the optimizer
train_op = tf.train.AdamOptimizer(1e-4).minimize(cross_entropy)
# Add the ops to initialize variables. These will include
# the optimizer slots added by AdamOptimizer().
init_op = tf.initialize_all_variables()
# launch the graph in a session
sess = tf.Session()
# Actually intialize the variables
sess.run(init_op)
# now train your model
for ...:
sess.run(train_op)
私は、adamオプティマイザーを使用するときに指数関数的減衰を使用すること、つまり次のコードを使用することが有用かどうか疑問に思っています。
...build the model...
# Add the optimizer
step = tf.Variable(0, trainable=False)
rate = tf.train.exponential_decay(0.15, step, 1, 0.9999)
optimizer = tf.train.AdamOptimizer(rate).minimize(cross_entropy, global_step=step)
# Add the ops to initialize variables. These will include
# the optimizer slots added by AdamOptimizer().
init_op = tf.initialize_all_variables()
# launch the graph in a session
sess = tf.Session()
# Actually intialize the variables
sess.run(init_op)
# now train your model
for ...:
sess.run(train_op)
通常、人々は何らかの学習率の減衰を使用します。アダムにとっては珍しいようです。これには理論的な理由はありますか?Adamオプティマイザーと減衰を組み合わせると便利ですか?
global_step
パラメーターを使用しますminimize
。編集を参照してください。
1e-4
= 0.0001
ではなく0.0004
。