Merge pull request #5005 from Pascal66/patch-2

Fix a rare error during hyperopt
This commit is contained in:
Matthias
2021-05-24 09:40:29 +01:00
committed by GitHub

View File

@@ -163,7 +163,8 @@ class Hyperopt:
:param epoch: result dictionary for this epoch.
"""
with self.results_file.open('a') as f:
rapidjson.dump(epoch, f, default=str, number_mode=rapidjson.NM_NATIVE)
rapidjson.dump(epoch, f, default=str,
number_mode=rapidjson.NM_NATIVE | rapidjson.NM_NAN)
f.write("\n")
self.num_epochs_saved += 1