def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--bert-model', default='bert-base', type=str)
    parser.add_argument('--input_dir', default=None, type=str, required=True, help='The input data dir. Should contain the .tsv files (or other data files) for the task.')
    parser.add_argument('--output_dir', default=None, type=str, required=True, help='The output directory where the model predictions and checkpoints will be written.')
    parser.add_argument('--max_seq_length', default=320, type=int, help='The maximum total input sequence length after WordPiece tokenization. \nSequences longer than this will be truncated, and sequences shorter \nthan this will be padded.')
    parser.add_argument('--seed', type=int, default=0, help='random seed for initialization')
    parser.add_argument('--doc_stride', type=int, default=128)
    parser.add_argument('--max_query_length', type=int, default=30)
    parser.add_argument('--max_answer_length', type=int, default=30)
    args = parser.parse_args()
    random.seed(args.seed)
    np.random.seed(args.seed)
    torch.manual_seed(args.seed)
    os.makedirs(args.output_dir, exist_ok=True)
    gen(args)
def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--bert-model', default='bert-base', type=str)
    parser.add_argument('--input_dir', default=None, type=str, required=True, help='The input data dir. Should contain the .tsv files (or other data files) for the task.')
    parser.add_argument('--output_dir', default=None, type=str, required=True, help='The output directory where the model predictions and checkpoints will be written.')
    parser.add_argument('--max_seq_length', default=320, type=int, help='The maximum total input sequence length after WordPiece tokenization. \nSequences longer than this will be truncated, and sequences shorter \nthan this will be padded.')
    parser.add_argument('--seed', type=int, default=0, help='random seed for initialization')
    parser.add_argument('--doc_stride', type=int, default=128)
    parser.add_argument('--max_query_length', type=int, default=30)
    parser.add_argument('--max_answer_length', type=int, default=30)
    args = parser.parse_args()
    random.seed(args.seed)
    np.random.seed(args.seed)
    torch.manual_seed(args.seed)
    os.makedirs(args.output_dir, exist_ok=True)
    gen(args)