Fixed issue related to float32 not being json serializable

zomux · Nov 28, 2018 · 4167d94 · 4167d94
1 parent f9de82c
commit 4167d94
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/nncompress/embed_compress.py b/nncompress/embed_compress.py
@@ -214,10 +214,10 @@ def train(self, embed_matrix, max_epochs=200):
                 logging.info(log_str)
                 dca_train_log.append(
                     {"epoch": epoch,
-                     "train_loss" : train_loss,
-                     "train_maxp" : train_maxp,
-                     "valid_loss" : valid_loss,
-                     "valid_maxp" : valid_maxp,
+                     "train_loss" : float(train_loss),
+                     "train_maxp" : float(train_maxp),
+                     "valid_loss" : float(valid_loss),
+                     "valid_maxp" : float(valid_maxp),
                      "batches_per_second" : batches_per_second,
                      "time_elapsed" : time_elapsed,
                      "report_token" : report_token}