From 35648ab1a21f3637bbde34404a1747f7f8cc43fc Mon Sep 17 00:00:00 2001 From: caoying03 Date: Mon, 4 Sep 2017 14:26:20 +0800 Subject: [PATCH] rename mse into square_error after dev branch update. --- ctr/network_conf.py | 2 +- dssm/network_conf.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ctr/network_conf.py b/ctr/network_conf.py index d577abcf..a90d1dc6 100644 --- a/ctr/network_conf.py +++ b/ctr/network_conf.py @@ -100,6 +100,6 @@ class CTRmodel(object): self.output = layer.fc( input=merge_layer, size=1, act=paddle.activation.Sigmoid()) if not self.is_infer: - self.train_cost = paddle.layer.mse_cost( + self.train_cost = paddle.layer.square_error_cost( input=self.output, label=self.click) return self.output diff --git a/dssm/network_conf.py b/dssm/network_conf.py index 996e91ae..04c2b7e2 100644 --- a/dssm/network_conf.py +++ b/dssm/network_conf.py @@ -270,7 +270,7 @@ class DSSM(object): input=prediction, label=label) else: prediction = paddle.layer.cos_sim(*semantics) - cost = paddle.layer.mse_cost(prediction, label) + cost = paddle.layer.square_error_cost(prediction, label) if not self.is_infer: return cost, prediction, label -- GitLab