def _init_session(self):
    config = tf1.ConfigProto(device_count={'gpu': 0})
    config.gpu_options.allow_growth = True
    return tf1.Session(config=config)
def _init_session(self):
    config = tf1.ConfigProto(device_count={'gpu': 0})
    config.gpu_options.allow_growth = True
    return tf1.Session(config=config)