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


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


Loading…
Cancel
Save