Browse Source

add config photo for LoadFromSavedModel

v0.15-tensorflow1.15
Nam Dinh Esther Hu 5 years ago
parent
commit
9b02c1bb81
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      src/TensorFlowNET.Core/Sessions/Session.cs

+ 2
- 2
src/TensorFlowNET.Core/Sessions/Session.cs View File

@@ -41,13 +41,13 @@ namespace Tensorflow
}

[MethodImpl(MethodImplOptions.NoOptimization)]
public static Session LoadFromSavedModel(string path)
public static Session LoadFromSavedModel(string path,string target="",ConfigProto config=null)
{
lock (Locks.ProcessWide)
{
var graph = c_api.TF_NewGraph();
var status = new Status();
var opt = new SessionOptions();
var opt = new SessionOptions(target,config);

var tags = new string[] {"serve"};
var buffer = new TF_Buffer();


Loading…
Cancel
Save