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

change default config for NewSessionCreator (#191)

parent d8627d63
......@@ -4,6 +4,7 @@
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import tensorflow as tf
from .common import get_default_sess_config
__all__ = ['NewSessionCreator', 'ReuseSessionCreator', 'SessionCreatorAdapter']
......@@ -13,8 +14,11 @@ class NewSessionCreator(tf.train.SessionCreator):
"""
Args:
target, graph, config: same as :meth:`Session.__init__()`.
config: defaults to :func:`tfutils.get_default_sess_config()`
"""
self.target = target
if config is None:
config = get_default_sess_config()
self.config = config
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