Ejemplo n.º 1
0
def create_parser(model, parser):
    parser = model.add_arguments(parser)
    parser = dataset.add_arguments(parser)
    parser = train.add_training_arguments(parser)
    parser = train.add_ipu_arguments(parser)
    parser = logging.add_arguments(parser)
    return parser
Ejemplo n.º 2
0
def create_parser(model, lr_schedule, parser):
    parser = model.add_arguments(parser)
    parser = dataset.add_arguments(parser)
    parser = add_training_arguments(parser)
    parser = lr_schedule.add_arguments(parser)
    parser = add_ipu_arguments(parser)
    parser = logging.add_arguments(parser)
    return parser
Ejemplo n.º 3
0
def create_all_options_parser():
    parser = argparse.ArgumentParser(
        description='BERT  Pretraining in TensorFlow',
        add_help=False,
        formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    parser = add_main_arguments(parser)
    parser = dataset.add_arguments(parser)
    parser = add_training_arguments(parser)
    parser = add_ipu_arguments(parser)
    parser = bert_logging.add_arguments(parser)
    parser = add_bert_arguments(parser)
    return parser