diff --git a/convlab/policy/emoTUS/evaluate.py b/convlab/policy/emoTUS/evaluate.py index 9d587e4a57eb2d71bb49b023a2d764b711eb43d9..2ce57dc8a27bd43679de3d3c64a0ab9ec7dad09e 100644 --- a/convlab/policy/emoTUS/evaluate.py +++ b/convlab/policy/emoTUS/evaluate.py @@ -45,12 +45,13 @@ class Evaluator: self.dataset = dataset self.model_checkpoint = model_checkpoint self.model_weight = model_weight - self.time = f"{datetime.now().strftime('%y-%m-%d-%H-%M')}" + self.time = f"{datetime.now().strftime('%y-%m-%d-%H-%M-%S')}" self.use_sentiment = kwargs.get("use_sentiment", False) self.add_persona = kwargs.get("add_persona", False) self.emotion_mid = kwargs.get("emotion_mid", False) self.emotion_weight = kwargs.get("weight", None) self.sample = kwargs.get("sample", False) + print("self.emotion_weight", self.emotion_weight) self.usr = UserActionPolicy( model_checkpoint,