未验证 提交 7e4d9df9 编写于 作者: L lilong12 提交者: GitHub

fix bug in example code for DGCMomentumOptimizer (#2246)

* modify the doc, test=develop
上级 d9a04a16
...@@ -73,6 +73,29 @@ DGC还使用动量因子掩藏(momentum factor masking)和预训练(warm-u ...@@ -73,6 +73,29 @@ DGC还使用动量因子掩藏(momentum factor masking)和预训练(warm-u
.. code-block:: python .. code-block:: python
import paddle.fluid as fluid import paddle.fluid as fluid
def network():
x = fluid.layers.data(name='x', shape=[1], dtype='int64', lod_level=0)
y = fluid.layers.data(name='y', shape=[1], dtype='int64', lod_level=0)
emb_x = fluid.layers.embedding(
input=x,
size=[10, 2],
is_sparse=False)
emb_y = fluid.layers.embedding(
input=y,
size=[10, 2],
is_sparse=False)
concat = fluid.layers.concat([emb_x, emb_y], axis=1)
fc = fluid.layers.fc(input=concat,
name="fc",
size=1,
num_flatten_dims=1,
bias_attr=False)
loss = fluid.layers.reduce_mean(fc)
return loss
loss = network() loss = network()
optimizer = fluid.optimizer.SGD(learning_rate=0.1) optimizer = fluid.optimizer.SGD(learning_rate=0.1)
params_grads = optimizer.backward(loss) params_grads = optimizer.backward(loss)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册