Skip to content

S
seminar-breakout
Commit 0ffd5337 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

avoid the unnecessary warning

parent adc9409f
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
tensorpack/predict/config.py
View file @ 0ffd5337
...@@ -6,7 +6,6 @@ import six ...@@ -6,7 +6,6 @@ import six
from ..compat import tfv1 as tf from ..compat import tfv1 as tf
from ..train.model_desc import ModelDescBase from ..train.model_desc import ModelDescBase
from ..tfutils import get_default_sess_config
from ..tfutils.sessinit import JustCurrentSession, SessionInit from ..tfutils.sessinit import JustCurrentSession, SessionInit
from ..tfutils.sesscreate import NewSessionCreator from ..tfutils.sesscreate import NewSessionCreator
from ..tfutils.tower import TowerFunc from ..tfutils.tower import TowerFunc
...@@ -100,7 +99,7 @@ class PredictConfig(object): ...@@ -100,7 +99,7 @@ class PredictConfig(object):
assert_type(self.session_init, SessionInit, 'session_init') assert_type(self.session_init, SessionInit, 'session_init')
if session_creator is None: if session_creator is None:
self.session_creator = NewSessionCreator(config=get_default_sess_config()) self.session_creator = NewSessionCreator()
else: else:
self.session_creator = session_creator self.session_creator = session_creator
......
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