fix sample code error of fluid.dygraph.NaturalExpDecay,test=develop (#25092)

fix sample code error of fluid.dygraph.NaturalExpDecay
fix-sync_batch_norm-hang-in-fleet
Zhou Wei 5 years ago committed by GitHub
parent 2996315fc9
commit a7cf562447
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -178,12 +178,14 @@ class NaturalExpDecay(LearningRateDecay):
import paddle.fluid as fluid
base_lr = 0.1
with fluid.dygraph.guard():
emb = fluid.dygraph.Embedding([10, 10])
sgd_optimizer = fluid.optimizer.SGD(
learning_rate=fluid.dygraph.NaturalExpDecay(
learning_rate=base_lr,
decay_steps=10000,
decay_rate=0.5,
staircase=True))
staircase=True),
parameter_list=emb.parameters())
"""

Loading…
Cancel
Save