INNER CODE UNIT · Python
main
yixinL7/BRIO · main.py:515
def main(args):
# set env
if len(args.gpuid) > 1:
os.environ['MASTER_ADDR'] = 'localhost'
os.environ['MASTER_PORT'] = f'{args.port}'
mp.spawn(run, args=(args,), nprocs=len(args.gpuid), join=True)
else:
run(0, args)
if __name__ == "__main__":
parser = argparse.ArgumentParser(description='Parameters')
parser.add_argument("--cuda", action="store_true", help="use cuda")
parser.add_argument("--gpuid", nargs='+', type=int, default=0, help="gpu ids")
parser.add_argument("-e", "--evaluate", action="store_true", help="evaluate model")
parser.add_argument("-r", "--do_reranking", action="store_true", help="do reranking evaluation")
parser.add_argument("-g", "--do_generation", action="store_true", help="do generation evaluation")
parser.add_argument("-l", "--log", action="store_true", help="logging")
parser.add_argument("-p", "--port", type=int, default=12355, help="port")