def setup_args():
"""
Setup arguments.
"""
parser = argparse.ArgumentParser()
models.add_cmdline_args(parser)
tasks.add_cmdline_args(parser)
parser.add_argument("--nsp_inference_model_path", type=str, required=True)
args = parse_args(parser)
args.load(args.config_path, "Model")
args.run_infer = True # only build infer program
print(json.dumps(args, indent=2))
return args
def setup_args():
"""
Setup arguments.
"""
parser = argparse.ArgumentParser()
parser.add_argument("--is_distributed", type=str2bool, default=False)
parser.add_argument("--save_path", type=str, default="output")
parser.add_argument("--infer_file", type=str, required=True)
parser.add_argument("--output_name", type=str, required=True)
parser.add_argument("--skip_steps", type=int, default=1)
models.add_cmdline_args(parser)
tasks.add_cmdline_args(parser)
args = parse_args(parser)
args.load(args.config_path, "Model")
args.run_infer = True # only build infer program
print(json.dumps(args, indent=2))
return args
def setup_args():
"""
Setup arguments.
"""
parser = argparse.ArgumentParser()
parser.add_argument("--is_distributed", type=str2bool, default=False)
parser.add_argument("--save_path", type=str, default="output")
parser.add_argument("--train_file", type=str, required=True)
parser.add_argument("--valid_file", type=str, required=True)
parser.add_argument("--num_epochs", type=int, default=20)
parser.add_argument("--log_steps", type=int, default=100)
parser.add_argument("--validation_steps", type=int, default=1000)
parser.add_argument("--save_steps", type=int, default=5000)
models.add_cmdline_args(parser)
tasks.add_cmdline_args(parser)
args = parse_args(parser)
args.load(args.config_path, "Model")
print(json.dumps(args, indent=2))
return args