Commit 083c1c5f authored by Yuxin Wu's avatar Yuxin Wu

change default config for NewSessionCreator (#191)

parent d8627d63
...@@ -4,6 +4,7 @@ ...@@ -4,6 +4,7 @@
# Author: Yuxin Wu <ppwwyyxxc@gmail.com> # Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import tensorflow as tf import tensorflow as tf
from .common import get_default_sess_config
__all__ = ['NewSessionCreator', 'ReuseSessionCreator', 'SessionCreatorAdapter'] __all__ = ['NewSessionCreator', 'ReuseSessionCreator', 'SessionCreatorAdapter']
...@@ -13,8 +14,11 @@ class NewSessionCreator(tf.train.SessionCreator): ...@@ -13,8 +14,11 @@ class NewSessionCreator(tf.train.SessionCreator):
""" """
Args: Args:
target, graph, config: same as :meth:`Session.__init__()`. target, graph, config: same as :meth:`Session.__init__()`.
config: defaults to :func:`tfutils.get_default_sess_config()`
""" """
self.target = target self.target = target
if config is None:
config = get_default_sess_config()
self.config = config self.config = config
self.graph = graph self.graph = graph
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment