未验证 提交 2435f3a2 编写于 作者: 1 123malin 提交者: GitHub

test=document, bug fix for LookaheadOptimizer and LarsMomentumOptimzer (#2249)

上级 f369cef6
...@@ -38,6 +38,7 @@ LarsMomentumOptimizer ...@@ -38,6 +38,7 @@ LarsMomentumOptimizer
.. code-block:: python .. code-block:: python
import paddle.fluid as fluid import paddle.fluid as fluid
import numpy as np
np_inp = np.array([[1.0, 2.0], [3.0, 4.0]], dtype=np.float32) np_inp = np.array([[1.0, 2.0], [3.0, 4.0]], dtype=np.float32)
inp = fluid.layers.data( inp = fluid.layers.data(
......
...@@ -29,7 +29,7 @@ LookaheadOptimizer ...@@ -29,7 +29,7 @@ LookaheadOptimizer
import paddle import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
import numpy as np import numpy.random as random
x = fluid.layers.data(name='x', shape=[2], dtype='float32') x = fluid.layers.data(name='x', shape=[2], dtype='float32')
label = fluid.layers.data(name="label", shape=[1], dtype="int64") label = fluid.layers.data(name="label", shape=[1], dtype="int64")
...@@ -46,11 +46,14 @@ LookaheadOptimizer ...@@ -46,11 +46,14 @@ LookaheadOptimizer
exe = fluid.Executor(place) exe = fluid.Executor(place)
exe.run(fluid.default_startup_program()) exe.run(fluid.default_startup_program())
def train_reader(limit=5):
for i in range(limit):
yield random.random([2]).astype('float32'), random.random([1]).astype('int64')
feeder = fluid.DataFeeder(feed_list=[x, label], place=place) feeder = fluid.DataFeeder(feed_list=[x, label], place=place)
reader = paddle.batch(paddle.reader.shuffle(train_reader, buf_size=50000),batch_size=1)
step = 0
while(step < 10): for batch_data in reader():
step += 1
exe.run(fluid.default_main_program(), exe.run(fluid.default_main_program(),
feed=feeder.feed(batch_data)) feed=feeder.feed(batch_data))
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册