INNER CODE UNIT · Python

main

AI4Bharat/Indic-BERT-v1 · fine_tune/cli.py:153

def main(argvec=None):
    parser = argparse.ArgumentParser()
    add_generic_args(parser, os.getcwd())
    for module in get_modules():
        module.add_model_specific_args(parser, os.getcwd())
    args = parser.parse_args(argvec)
    hparams = vars(args)

    # high-level command line parameters
    dataset = hparams['dataset']
    train_lang = hparams.get('train_lang', hparams['lang'])
    test_lang = hparams['lang']
    model = hparams['model']
    iglue_dir = hparams['iglue_dir']

    data_dir = os.path.join(iglue_dir, dataset)
    output_dir = os.path.join(hparams['output_dir'], dataset,
                              'train-{}'.format(train_lang),

View source record →