From 6c20953313ef71e86185c85e533f0888ac98d49f Mon Sep 17 00:00:00 2001 From: pkpk Date: Mon, 24 Jun 2019 20:27:10 +0800 Subject: [PATCH] test=develop (#2524) --- .../dialogue_general_understanding/define_paradigm.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/PaddleNLP/models/dialogue_model_toolkit/dialogue_general_understanding/define_paradigm.py b/PaddleNLP/models/dialogue_model_toolkit/dialogue_general_understanding/define_paradigm.py index 07581b59..63d1952e 100644 --- a/PaddleNLP/models/dialogue_model_toolkit/dialogue_general_understanding/define_paradigm.py +++ b/PaddleNLP/models/dialogue_model_toolkit/dialogue_general_understanding/define_paradigm.py @@ -62,7 +62,7 @@ class Paradigm(object): ce_loss, probs = fluid.layers.softmax_with_cross_entropy( logits=logits, label=params['labels'], return_softmax=True) - loss = fluid.layers.mean(input=ce_loss) + loss = fluid.layers.mean(x=ce_loss) num_seqs = fluid.layers.create_tensor(dtype='int64') accuracy = fluid.layers.accuracy( input=probs, label=params['labels'], total=num_seqs) @@ -102,7 +102,7 @@ class Paradigm(object): ce_loss = fluid.layers.reduce_sum( fluid.layers.sigmoid_cross_entropy_with_logits( x=logits, label=labels_onehot)) - loss = fluid.layers.mean(input=ce_loss) + loss = fluid.layers.mean(x=ce_loss) probs = fluid.layers.sigmoid(logits) if params['is_prediction']: @@ -158,7 +158,7 @@ class Paradigm(object): correct_prediction, dtype='float32')) ce_loss = fluid.layers.softmax_with_cross_entropy(logits=logits, \ label=fluid.layers.reshape(params['labels'], [-1, 1])) - loss = fluid.layers.mean(input=ce_loss) + loss = fluid.layers.mean(x=ce_loss) loss.persistable = True probs.persistable = True -- GitLab