diff --git a/src/TensorFlowNET.Core/Keras/backend.cs b/src/TensorFlowNET.Core/Keras/backend.cs index 5ac04f1a..46769bd8 100644 --- a/src/TensorFlowNET.Core/Keras/backend.cs +++ b/src/TensorFlowNET.Core/Keras/backend.cs @@ -28,7 +28,7 @@ namespace Tensorflow.Keras //Func py_any = any; //Func> py_slice = slice; - public static Session _SESSION = tf.defaultSession; + public static Session _SESSION = ops.get_default_session(); public static Graph _GRAPH = null; public static Dictionary _GRAPH_LEARNING_PHASES; //Dictionary> PER_GRAPH_LAYER_NAME_UIDS; diff --git a/src/TensorFlowNET.Core/tensorflow.cs b/src/TensorFlowNET.Core/tensorflow.cs index 8fade290..cf973864 100644 --- a/src/TensorFlowNET.Core/tensorflow.cs +++ b/src/TensorFlowNET.Core/tensorflow.cs @@ -40,7 +40,7 @@ namespace Tensorflow public tensorflow() { - _defaultSessionFactory = new ThreadLocal(Session); + _defaultSessionFactory = new ThreadLocal(() => new Session()); } public Session defaultSession => _defaultSessionFactory.Value;