init

local_coh_ppl.py

@@ -247,11 +247,10 @@ class LocalCohPPL(evaluate.Measurement):
             #
             all_norm_ppl.extend(norm_ppl)
 
-            for x in norm_ppl:
-                if np.isnan(x):
-                    print("[compute ppl] nan ...")
-                    pdb.set_trace()
-                    print(">>")
+            if any(np.isnan(norm_ppl)):
+                print("[compute ppl] nan ...")
+                pdb.set_trace()
+                print(">>")
 
         #
         avg_ppl = []
@@ -259,4 +258,9 @@ class LocalCohPPL(evaluate.Measurement):
         for _len in blens:
             avg_ppl.append( float(np.mean(all_norm_ppl[offset:offset+_len])) )
             offset += _len
+        if any(np.isnan(avg_ppl)):
+            print("[compute ppl] nan ...")
+            pdb.set_trace()
+            print(">>")
+
         return {"local_coh_ppl": avg_ppl}