| /external/tensorflow/tensorflow/python/training/ |
| rmsprop_test.py | 40 # learning_rate, decay, momentum, epsilon, centered, use_resource
92 for (dtype, learning_rate, decay, momentum,
110 learning_rate=learning_rate,
148 var0_np, grads0_np, mg0_np, rms0_np, mom0_np, learning_rate,
151 var1_np, grads1_np, mg1_np, rms1_np, mom1_np, learning_rate,
173 learning_rate=1.0,
195 learning_rate=1.0,
211 for (dtype, learning_rate, decay,
231 learning_rate=learning_rate
[all...] |
| momentum_test.py | 59 learning_rate = lambda: 2.0
62 learning_rate = learning_rate()
65 learning_rate=learning_rate, momentum=momentum)
86 # update: v -= grad * learning_rate
183 learning_rate=2.0, momentum=0.9, use_nesterov=True)
219 learning_rate=2.0, momentum=0.9, use_nesterov=True)
249 opt = momentum_lib.MomentumOptimizer(learning_rate=1.0, momentum=0.0)
264 opt = momentum_lib.MomentumOptimizer(learning_rate=1.0, momentum=0.0
[all...] |
| adadelta.py | 36 def __init__(self, learning_rate=0.001, rho=0.95, epsilon=1e-8,
41 learning_rate: A `Tensor` or a floating point value. The learning rate.
51 self._lr = learning_rate
|
| adagrad.py | 40 def __init__(self, learning_rate, initial_accumulator_value=0.1,
45 learning_rate: A `Tensor` or a floating point value. The learning rate.
59 self._learning_rate = learning_rate
82 name="learning_rate")
|
| proximal_adagrad.py | 37 def __init__(self, learning_rate, initial_accumulator_value=0.1,
43 learning_rate: A `Tensor` or a floating point value. The learning rate.
61 self._learning_rate = learning_rate
80 name="learning_rate")
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/examples/ |
| binary_mnist.py | 22 --output_dir="/tmp/binary_mnist" --depth=4 --learning_rate=0.3 \
79 learner_config.learning_rate_tuner.fixed.learning_rate = FLAGS.learning_rate
151 "--learning_rate",
|
| boston.py | 24 --batch_size=404 --output_dir="/tmp/boston" --depth=4 --learning_rate=0.1 \
56 learner_config.learning_rate_tuner.fixed.learning_rate = FLAGS.learning_rate
156 "--learning_rate",
|
| mnist.py | 23 --output_dir="/tmp/mnist" --depth=4 --learning_rate=0.3 --batch_size=60000 \
76 learner_config.learning_rate_tuner.fixed.learning_rate = FLAGS.learning_rate
153 "--learning_rate",
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/linear_regression/ |
| linear_regression.py | 138 learning_rate = 0.1
149 optimizer = tf.train.GradientDescentOptimizer(learning_rate)
|
| linear_regression_test.py | 75 optimizer = tf.train.GradientDescentOptimizer(learning_rate=0.1)
100 optimizer = tf.train.GradientDescentOptimizer(learning_rate=0.1)
|
| linear_regression_graph_test.py | 54 learning_rate=0.1).minimize(loss)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| auto_parallel_test.cc | 41 Output learning_rate = ops::Const(s.WithOpName("learning_rate"), 0.01f, {1}); local
43 s.WithOpName("apply_gradient"), {var}, {learning_rate}, {add});
94 EXPECT_EQ("AutoParallel-Replica-0/learning_rate", node_learning_rate0.name());
117 EXPECT_EQ("AutoParallel-Replica-1/learning_rate", node_learning_rate1.name());
|
| /external/tensorflow/tensorflow/examples/image_retraining/ |
| retrain_test.py | 69 @tf.test.mock.patch.object(retrain, 'FLAGS', learning_rate=0.01)
78 @tf.test.mock.patch.object(retrain, 'FLAGS', learning_rate=0.01)
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/mnist/ |
| mnist_test.py | 46 optimizer = tf.train.GradientDescentOptimizer(learning_rate=0.01)
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_colorbot/ |
| rnn_colorbot_test.py | 54 optimizer = tf.train.AdamOptimizer(learning_rate=.01)
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/python/layers/ |
| decisions_to_data_test.py | 46 learning_rate=0.01,
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| momentum_test.py | 50 learning_rate=2.0, momentum=0.9)
67 # update: v -= grad * learning_rate
109 learning_rate=0.1, momentum=0.9, use_nesterov=True)
129 learning_rate=constant_op.constant(2.0),
147 # update: v -= grad * learning_rate
|
| /external/tensorflow/tensorflow/contrib/bayesflow/python/ops/ |
| sgld_optimizer.py | 50 learning_rate: Scalar `float`-like `Tensor`. The base learning rate for the
79 learning_rate,
88 learning_rate, preconditioner_decay_rate, num_pseudo_batches, burnin,
107 learning_rate, name='learning_rate')
157 self._learning_rate, message='`learning_rate` must be non-negative')
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/ |
| training_ops_test.py | 62 config.learning_rate_tuner.dropout.learning_rate = dropout_learning_rate
318 learning_rate=0.1,
469 learning_rate=0.1,
654 learning_rate=0.2,
793 learning_rate=0.1,
862 learning_rate=0.1,
966 learning_rate=0.1
[all...] |
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| moving_average_optimizer_test.py | 64 gradient_descent.GradientDescentOptimizer(learning_rate=2.0),
171 gradient_descent.GradientDescentOptimizer(learning_rate=2.0))
193 wrapper_opt = WrapperOptimizer(learning_rate=2.0)
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/python/models/ |
| forest_to_data_then_nn_test.py | 49 learning_rate=0.01,
|
| k_feature_decisions_to_data_then_nn_test.py | 49 learning_rate=0.01,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| gbdt_batch_test.py | 151 learner_config.learning_rate_tuner.fixed.learning_rate = 0.1
254 learner_config.learning_rate_tuner.fixed.learning_rate = 0.1
358 learner_config.learning_rate_tuner.fixed.learning_rate = 0.1
429 learner_config.learning_rate_tuner.fixed.learning_rate = 0.1
492 learner_config.learning_rate_tuner.fixed.learning_rate = 0.1
576 learner_config.learning_rate_tuner.fixed.learning_rate = 0.1
608 learner_config.learning_rate_tuner.fixed.learning_rate = 1
711 learner_config.learning_rate_tuner.fixed.learning_rate = 1
[all...] |
| /external/tensorflow/tensorflow/contrib/eager/python/examples/resnet50/ |
| resnet50_graph_test.py | 84 optimizer = tf.train.GradientDescentOptimizer(learning_rate=0.01)
147 optimizer = tf.train.GradientDescentOptimizer(learning_rate=1.0)
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_ptb/ |
| rnn_ptb_graph_test.py | 45 optimizer = tf.train.GradientDescentOptimizer(learning_rate=1.0)
130 optimizer = tf.train.GradientDescentOptimizer(learning_rate=1.0)
|
