Browse Source

Revise some details.

tags/v0.100.5-BERT-load
Yaohui Liu 2 years ago
parent
commit
2f62caa4e5
No known key found for this signature in database GPG Key ID: E86D01E1809BD23E
2 changed files with 3 additions and 2 deletions
  1. +2
    -1
      src/TensorFlowNET.Keras/Saving/SavedModel/load.cs
  2. +1
    -1
      test/TensorFlowNET.Keras.UnitTest/SaveModel/SequentialModelLoad.cs

+ 2
- 1
src/TensorFlowNET.Keras/Saving/SavedModel/load.cs View File

@@ -56,7 +56,8 @@ namespace Tensorflow.Keras.Saving.SavedModel
}
else
{
throw new NotImplementedException("Not implemented, please submit an issue to https://github.com/SciSharp/TensorFlow.NET/issues.");
throw new NotImplementedException("SavedModel saved prior to TF 2.5 detected when loading Keras model, please" +
" use higher version or submit an issue to https://github.com/SciSharp/TensorFlow.NET/issues. to let us know you need it.");
}

if (metadata.Nodes is null || metadata.Nodes.Count == 0)


+ 1
- 1
test/TensorFlowNET.Keras.UnitTest/SaveModel/SequentialModelLoad.cs View File

@@ -59,7 +59,7 @@ public class SequentialModelLoad
}

[TestMethod]
public void Temp()
public void ModelWithSelfDefinedModule()
{
var model = tf.keras.models.load_model(@"Assets/python_func_model");
model.summary();


Loading…
Cancel
Save