remove some debug message

convlab/policy/emoUS/emotion_eval.py

@@ -326,6 +326,7 @@ def main():
                      emotion_mid=args.emotion_mid,
                      weight=args.weight,
                      sample=args.sample)
+    print("=== evaluation ===")
     print("model checkpoint", args.model_checkpoint)
     print("generated_file", args.generated_file)
     print("input_file", args.input_file)

convlab/policy/emoUS/evaluate.py

@@ -98,7 +98,7 @@ class Evaluator:
         if self.sample:
             mode = "sample"
 
-        for dialog in tqdm(in_file['dialog']):
+        for dialog in tqdm(in_file['dialog'][:1]):
             inputs = dialog["in"]
             labels = self.usr._parse_output(dialog["out"])
 
@@ -341,6 +341,7 @@ def main():
                      emotion_mid=args.emotion_mid,
                      weight=args.weight,
                      sample=args.sample)
+    print("=== evaluation ===")
     print("model checkpoint", args.model_checkpoint)
     print("generated_file", args.generated_file)
     print("input_file", args.input_file)

convlab/policy/genTUS/ppo/vector.py

@@ -19,8 +19,8 @@ class stepGenTUSVector:
         self.mentioned_domain = []
         self.allow_general_intent = allow_general_intent
         self.candidate_num = 5
-        if self.allow_general_intent:
-            print("---> allow_general_intent")
+        # if self.allow_general_intent:
+        #     print("---> allow_general_intent")
 
     def init_session(self, goal: Goal):
         self.goal = goal

convlab/policy/genTUS/stepGenTUS.py

@@ -32,7 +32,7 @@ class UserActionPolicy(Policy):
         self.max_in_len = 500
         self.max_out_len = 100 if only_action else 200
         max_act_len = kwargs.get("max_act_len", 2)
-        print("max_act_len", max_act_len)
+        # print("max_act_len", max_act_len)
         self.max_action_len = max_act_len
         if "max_act_len" in kwargs:
             self.max_out_len = 30 * self.max_action_len

convlab/policy/genTUS/unify/knowledge_graph.py

@@ -11,7 +11,7 @@ DATASET = "unify"
 
 class KnowledgeGraph:
     def __init__(self, tokenizer: BartTokenizer, ontology_file=None, dataset="multiwoz21"):
-        print("dataset", dataset)
+        # print("dataset", dataset)
         self.debug = DEBUG
         self.tokenizer = tokenizer