Browse Source

update:hdf5_format -> WriteAttrs

tags/yolov3
dataangel Haiping 4 years ago
parent
commit
536b534c36
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      src/TensorFlowNET.Keras/Saving/hdf5_format.cs

+ 4
- 4
src/TensorFlowNET.Keras/Saving/hdf5_format.cs View File

@@ -140,7 +140,7 @@ namespace Tensorflow.Keras.Saving
if (get_Name.IndexOf("/") > 1) {
get_Name = get_Name.Split('/')[1];
}
(bool success, Array result) = Hdf5.ReadDataset<float>(g, get_Name);
(bool success, Array result) = Hdf5.ReadDataset<float>(g, get_Name, alternativeName: i_);
if (success)
weight_values.Add(np.array(result));
}
@@ -276,13 +276,13 @@ namespace Tensorflow.Keras.Saving
Hdf5.WriteDatasetFromArray<float>(f, name, data.numpy().ToMuliDimArray<float>());
break;
case TF_DataType.TF_DOUBLE:
Hdf5.WriteDatasetFromArray<double>(f, name, data.numpy().ToMuliDimArray<float>());
Hdf5.WriteDatasetFromArray<double>(f, name, data.numpy().ToMuliDimArray<double>());
break;
case TF_DataType.TF_INT32:
Hdf5.WriteDatasetFromArray<int>(f, name, data.numpy().ToMuliDimArray<float>());
Hdf5.WriteDatasetFromArray<int>(f, name, data.numpy().ToMuliDimArray<int>());
break;
case TF_DataType.TF_INT64:
Hdf5.WriteDatasetFromArray<long>(f, name, data.numpy().ToMuliDimArray<float>());
Hdf5.WriteDatasetFromArray<long>(f, name, data.numpy().ToMuliDimArray<long>());
break;
default:
Hdf5.WriteDatasetFromArray<float>(f, name, data.numpy().ToMuliDimArray<float>());


Loading…
Cancel
Save