|
|
|
@ -181,7 +181,10 @@ def train_main(use_cuda, is_sparse, is_local=True):
|
|
|
|
|
cost = pd.cross_entropy(input=rnn_out, label=label)
|
|
|
|
|
avg_cost = pd.mean(cost)
|
|
|
|
|
|
|
|
|
|
optimizer = fluid.optimizer.Adagrad(learning_rate=1e-4)
|
|
|
|
|
optimizer = fluid.optimizer.Adagrad(
|
|
|
|
|
learning_rate=1e-4,
|
|
|
|
|
regularization=fluid.regularizer.L2DecayRegularizer(
|
|
|
|
|
regularization_coeff=0.1))
|
|
|
|
|
optimize_ops, params_grads = optimizer.minimize(avg_cost)
|
|
|
|
|
|
|
|
|
|
train_data = paddle.batch(
|
|
|
|
|