ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-06 01:18:52 -06:00
Code Issues Packages Projects Releases Wiki Activity

Hdf5Sample: Store all fp arrays as float32

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-08-12 07:51:59 -05:00
parent 53a7c8f84a
commit 2b00cf5b96
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
miplearn/features/sample.py
View File

@@ -178,8 +178,8 @@ class Hdf5Sample(Sample):
if value is None:
return
self._assert_is_array(value)
if len(value.shape) > 1 and value.dtype.kind == "f":
value = value.astype("float16")
if value.dtype.kind == "f":
value = value.astype("float32")
if key in self.file:
del self.file[key]
return self.file.create_dataset(key, data=value, compression="gzip")