Hdf5Sample: Store all fp arrays as float32

master
Alinson S. Xavier 4 years ago
parent 53a7c8f84a
commit 2b00cf5b96

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

Loading…
Cancel
Save