Skip to content
Snippets Groups Projects
Commit 55753fb9 authored by Hsien-Chin Lin's avatar Hsien-Chin Lin
Browse files

change default

parent cf02ef68
Branches
No related tags found
No related merge requests found
...@@ -17,8 +17,12 @@ DEBUG = False ...@@ -17,8 +17,12 @@ DEBUG = False
class UserActionPolicy(GenTUSUserActionPolicy): class UserActionPolicy(GenTUSUserActionPolicy):
def __init__(self, model_checkpoint, mode="language", only_action=False, max_turn=40, **kwargs): def __init__(self, model_checkpoint, mode="language", only_action=False, max_turn=40, **kwargs):
self.use_sentiment = kwargs.get("use_sentiment", False) 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) 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) super().__init__(model_checkpoint, mode, only_action, max_turn, **kwargs)
weight = kwargs.get("weight", None) weight = kwargs.get("weight", None)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment