def main():
parser = argparse.ArgumentParser()
parser.add_argument("-t",
"--tokens",
type=int,
default=DEFAULT_NUM_TOKENS,
help="number of tokens")
parser.add_argument("-p",
"--port",
type=int,
default=DEFAULT_PORT,
help="Set port number")
args = parser.parse_args()
runner.run(LEGOCognitiveEngine, ENGINE_NAME, INPUT_QUEUE_MAX_SIZE,
args.port, args.tokens)
def main():
parser = argparse.ArgumentParser()
parser.add_argument(
"-t", "--tokens", type=int, default=DEFAULT_NUM_TOKENS,
help="number of tokens")
parser.add_argument(
"-c",
"--cpu-only",
action="store_true",
help="Pass this flag to prevent the GPU from being used.")
parser.add_argument(
"-p", "--port", type=int, default=DEFAULT_PORT, help="Set port number")
args = parser.parse_args()
def engine_setup():
return SandwichEngine(args.cpu_only)
runner.run(
engine_setup, ENGINE_NAME, INPUT_QUEUE_MAX_SIZE, args.port, args.tokens)
def main():
parser = argparse.ArgumentParser()
parser.add_argument("-t",
"--tokens",
type=int,
default=DEFAULT_NUM_TOKENS,
help="number of tokens")
parser.add_argument("-p",
"--port",
type=int,
default=DEFAULT_PORT,
help="Set port number")
args = parser.parse_args()
def engine_setup():
return PingpongEngine()
runner.run(engine_setup, ENGINE_NAME, INPUT_QUEUE_MAX_SIZE, args.port,
args.tokens)
def main(tokens=DEFAULT_NUM_TOKENS, port=DEFAULT_PORT, use_gpu=True):
def engine_setup():
return RiblocEngine(use_gpu)
runner.run(
engine_setup, ENGINE_NAME, INPUT_QUEUE_MAX_SIZE, port, tokens)