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
commit 5d44ca0b82
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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