From d6de891e13c986b63c72595b3732b82bbc88172c Mon Sep 17 00:00:00 2001 From: qijun Date: Thu, 19 Oct 2017 16:40:50 -0700 Subject: [PATCH] add gitignore --- python/paddle/v2/framework/tests/.gitignore | 1 + python/paddle/v2/framework/tests/test_fit_a_line.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) create mode 100644 python/paddle/v2/framework/tests/.gitignore diff --git a/python/paddle/v2/framework/tests/.gitignore b/python/paddle/v2/framework/tests/.gitignore new file mode 100644 index 00000000000..28433306d49 --- /dev/null +++ b/python/paddle/v2/framework/tests/.gitignore @@ -0,0 +1 @@ +image/ diff --git a/python/paddle/v2/framework/tests/test_fit_a_line.py b/python/paddle/v2/framework/tests/test_fit_a_line.py index c4e645a2c46..f7b8c525779 100644 --- a/python/paddle/v2/framework/tests/test_fit_a_line.py +++ b/python/paddle/v2/framework/tests/test_fit_a_line.py @@ -33,10 +33,10 @@ cost = layers.square_error_cost( input=y_predict, label=y, program=program, init_program=init_program) avg_cost = layers.mean(x=cost, program=program, init_program=init_program) -sgd_optimizer = optimizer.SGDOptimizer(learning_rate=0.005) +sgd_optimizer = optimizer.SGDOptimizer(learning_rate=0.0001) opts = sgd_optimizer.minimize(avg_cost) -BATCH_SIZE = 10 +BATCH_SIZE = 2 train_reader = paddle.batch( paddle.reader.shuffle( -- GitLab