From 28e482239f0d3dff97781ed0960c0e5f7e310473 Mon Sep 17 00:00:00 2001
From: ceci3 <592712189@qq.com>
Date: Thu, 23 Apr 2020 08:18:55 +0000
Subject: [PATCH] fix lstm loss

---
 paddleslim/common/RL_controller/LSTM/LSTM_Controller.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/paddleslim/common/RL_controller/LSTM/LSTM_Controller.py b/paddleslim/common/RL_controller/LSTM/LSTM_Controller.py
index 30dd2907..521c6f4b 100644
--- a/paddleslim/common/RL_controller/LSTM/LSTM_Controller.py
+++ b/paddleslim/common/RL_controller/LSTM/LSTM_Controller.py
@@ -193,8 +193,7 @@ class LSTM(RLBaseController):
 
             fluid.layers.assign(self.baseline - (1.0 - self.decay) *
                                 (self.baseline - self.rewards), self.baseline)
-            self.loss = -1.0 * self.sample_log_probs * (
-                self.rewards - self.baseline)
+            self.loss = self.sample_log_probs * (self.rewards - self.baseline)
             fluid.clip.set_gradient_clip(
                 clip=fluid.clip.GradientClipByGlobalNorm(clip_norm=5.0))
             optimizer = fluid.optimizer.Adam(learning_rate=1e-3)
-- 
GitLab