Only initialize cider_score at the first time.

ruotianluo · Oct 12, 2017 · fd445dc · fd445dc
1 parent 52ec5c8
commit fd445dc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/misc/rewards.py b/misc/rewards.py
@@ -18,7 +18,7 @@
 
 def init_cider_scorer(cached_tokens):
     global CiderD_scorer
-    CiderD_scorer = CiderD(df=cached_tokens)
+    CiderD_scorer = CiderD_scorer or CiderD(df=cached_tokens)
 
 def array_to_str(arr):
     out = ''
@@ -59,4 +59,4 @@ def get_self_critical_reward(model, fc_feats, att_feats, data, gen_result):
 
     rewards = np.repeat(scores[:, np.newaxis], gen_result.shape[1], 1)
 
-    return rewards
+    return rewards