You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
System information
Have I written custom code (as opposed to using a stock example script provided in TensorFlow): NO
OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Ubuntu
TensorFlow installed from (source or binary): source
TensorFlow version (use command below): Master
Python version: 3.6
Bazel version (if compiling from source): N/A
GCC/Compiler version (if compiling from source): N/A
CUDA/cuDNN version: None
GPU model and memory: None
Exact command to reproduce: Run the snippet below.
In the Java, only can set the run options and can't modify the session configuration when load the model. Sometimes I want to set the session strategies, ex. Set the session threads, op threads number, and so on.
I write the code to add new interface in the JAVA, and also add the JNI corresponding method.
I have been submit the code in the github:
URL: https://github.com/raintung/TensorflowPatch
I will use it in my company project, I think it should not conflict the Tensorflow license, I still contribute it.
The text was updated successfully, but these errors were encountered:
If I understand correctly, you're proposing that a variant of SavedModelBundle.load() be added that takes in the session configuration (e.g., a serialized ConfigProto protocol buffer)?
Sure, that sounds perfectly reasonable.
Marking this as contributions welcome, feel free to send in a pull request. And I would suggest that we don't add a new native method, simply modify the existing one in place.
System information
Have I written custom code (as opposed to using a stock example script provided in TensorFlow): NO
OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Ubuntu
TensorFlow installed from (source or binary): source
TensorFlow version (use command below): Master
Python version: 3.6
Bazel version (if compiling from source): N/A
GCC/Compiler version (if compiling from source): N/A
CUDA/cuDNN version: None
GPU model and memory: None
Exact command to reproduce: Run the snippet below.
In the Java, only can set the run options and can't modify the session configuration when load the model. Sometimes I want to set the session strategies, ex. Set the session threads, op threads number, and so on.
I write the code to add new interface in the JAVA, and also add the JNI corresponding method.
I have been submit the code in the github:
URL:
https://github.com/raintung/TensorflowPatch
I will use it in my company project, I think it should not conflict the Tensorflow license, I still contribute it.
The text was updated successfully, but these errors were encountered: