17.

scrl/rewards.py

@@ -389,6 +389,7 @@ class RougeReward:
 
 class AmrReward:
     name = "AmrReward"
+    safe_amr = {}
     nlp = util.init_amr(STOG_MODEL_TRAIN)
 
     def __init__(self, weight=1, device="None"):
@@ -399,6 +400,11 @@ class AmrReward:
         scores = []
         for source, summary in zip(sources, summaries):
             if source != "" and summary != "":
+                if source in self.safe_amr:
+                    graph_source = self.safe_amr[source]
+                else:
+                    graph_source = util.calc_one_amr(self.nlp, source, "train_source")
+                    self.safe_amr[source] = graph_source
 
                 graph_source = util.calc_one_amr(self.nlp, source, "train_source")
                 graph_summary = util.calc_one_amr(self.nlp, summary, "train_summary")