From 55753fb9830c74b09d28457b61bdb0b809b61dc0 Mon Sep 17 00:00:00 2001
From: Hsien-Chin Lin <linh@hhu.de>
Date: Wed, 25 Jan 2023 12:01:14 +0100
Subject: [PATCH] change default
---
convlab/policy/emoTUS/emoTUS.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/convlab/policy/emoTUS/emoTUS.py b/convlab/policy/emoTUS/emoTUS.py
index 8590317d..dadfc923 100644
--- a/convlab/policy/emoTUS/emoTUS.py
+++ b/convlab/policy/emoTUS/emoTUS.py
@@ -17,8 +17,12 @@ DEBUG = False
class UserActionPolicy(GenTUSUserActionPolicy):
def __init__(self, model_checkpoint, mode="language", only_action=False, max_turn=40, **kwargs):
self.use_sentiment = kwargs.get("use_sentiment", False)
- self.add_persona = kwargs.get("add_persona", False)
+ self.add_persona = kwargs.get("add_persona", True)
self.emotion_mid = kwargs.get("emotion_mid", False)
+ print("===== model status =====")
+ print("use_sentiment: ", self.use_sentiment)
+ print("add_persona: ", self.add_persona)
+ print("emotion_mid: ", self.emotion_mid)
super().__init__(model_checkpoint, mode, only_action, max_turn, **kwargs)
weight = kwargs.get("weight", None)
--
GitLab