Fix the perplexity value

emedvedev · Sep 28, 2017 · 74d2dfd · 74d2dfd
1 parent 235225e
commit 74d2dfd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/aocr/model/model.py b/aocr/model/model.py
@@ -345,6 +345,7 @@ def train(self):
                 step_time, loss = 0.0, 0.0
 
         # Print statistics for the previous epoch.
+        perplexity = math.exp(loss) if loss < 300 else float('inf')
         logging.info("Global step %d. Time: %.3f, loss: %f, perplexity: %.2f."
                      % (self.sess.run(self.global_step), step_time, loss, perplexity))
         # Save checkpoint and reset timer and loss.