(config_dict: dict = None, force_as_default: bool = False)
return session
Create tf.Session based on config dict.
Create tf.Session based on config dict.
Create tf . Session based on config dict .
def create_session(config_dict: dict = None, force_as_default: bool = False) -> tf.Session:
    """Create tf.Session based on config dict."""
    # Setup TensorFlow config proto.
    cfg = _sanitize_tf_config(config_dict)
    config_proto = tf.ConfigProto()
    for key, value in cfg.items():
        fields = key.split(".")
        if fields[0] not in ["rnd", "env"]:
            obj = config_proto
            for field in fields[:-1]:
                obj = getattr(obj, field)
            setattr(obj, fields[-1], value)

    # Create session.
    session = tf.Session(config=config_proto)
    if force_as_default:
        # pylint: disable=protected-access
        session._default_session = session.as_default()
        session._default_session.enforce_nesting = False
        session._default_session.__enter__()
    return session
def create_session ( config_dict : dict = None , force_as_default : bool = False ) -> tf . Session : # Setup TensorFlow config proto. cfg = _sanitize_tf_config ( config_dict ) config_proto = tf . ConfigProto ( ) for key , value in cfg . items ( ) : fields = key . split ( "." ) if fields [ 0 ] not in [ "rnd" , "env" ] : obj = config_proto for field in fields [ : - 1 ] : obj = getattr ( obj , field ) setattr ( obj , fields [ - 1 ] , value ) # Create session. session = tf . Session ( config = config_proto ) if force_as_default : # pylint: disable=protected-access session . _default_session = session . as_default ( ) session . _default_session . enforce_nesting = False session . _default_session . __enter__ ( ) return session