diff --git a/python/paddle/v2/framework/tests/.gitignore b/python/paddle/v2/framework/tests/.gitignore new file mode 100644 index 0000000000000000000000000000000000000000..28433306d49112cc860f4ace9efca2b2d70deb3f --- /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 c4e645a2c465c64e50077dc56b01ac07a1c79202..f7b8c5257791e74db4d6e31e73bac136c4156a74 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(