Skip to content

Commit 536b534

Browse files
dataangelOceania2018
authored andcommitted
update:hdf5_format -> WriteAttrs
1 parent 2f3a17d commit 536b534

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/TensorFlowNET.Keras/Saving/hdf5_format.cs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ public static void load_weights_from_hdf5_group(long f, List<ILayer> layers)
140140
if (get_Name.IndexOf("/") > 1) {
141141
get_Name = get_Name.Split('/')[1];
142142
}
143-
(bool success, Array result) = Hdf5.ReadDataset<float>(g, get_Name);
143+
(bool success, Array result) = Hdf5.ReadDataset<float>(g, get_Name, alternativeName: i_);
144144
if (success)
145145
weight_values.Add(np.array(result));
146146
}
@@ -276,13 +276,13 @@ private static void WriteDataset(long f, string name, Tensor data)
276276
Hdf5.WriteDatasetFromArray<float>(f, name, data.numpy().ToMuliDimArray<float>());
277277
break;
278278
case TF_DataType.TF_DOUBLE:
279-
Hdf5.WriteDatasetFromArray<double>(f, name, data.numpy().ToMuliDimArray<float>());
279+
Hdf5.WriteDatasetFromArray<double>(f, name, data.numpy().ToMuliDimArray<double>());
280280
break;
281281
case TF_DataType.TF_INT32:
282-
Hdf5.WriteDatasetFromArray<int>(f, name, data.numpy().ToMuliDimArray<float>());
282+
Hdf5.WriteDatasetFromArray<int>(f, name, data.numpy().ToMuliDimArray<int>());
283283
break;
284284
case TF_DataType.TF_INT64:
285-
Hdf5.WriteDatasetFromArray<long>(f, name, data.numpy().ToMuliDimArray<float>());
285+
Hdf5.WriteDatasetFromArray<long>(f, name, data.numpy().ToMuliDimArray<long>());
286286
break;
287287
default:
288288
Hdf5.WriteDatasetFromArray<float>(f, name, data.numpy().ToMuliDimArray<float>());

0 commit comments

Comments
 (0)