Skip to content

S
seminar-breakout
You need to sign in or sign up before continuing.
Commit 0ffd5337 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

avoid the unnecessary warning

parent adc9409f
Hide 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