INNER CODE UNIT · Python
add_generic_args
AI4Bharat/Indic-BERT-v1 · fine_tune/cli.py:31
def add_generic_args(parser, root_dir):
# task-specific args START
parser.add_argument(
'--dataset',
type=str,
required=True,
help='The evaluation dataset to use'
)
parser.add_argument(
'--lang',
default=None,
type=str,
required=True,
help='ISO code of test language',
)
parser.add_argument(
'--train_lang',