Ejemplo n.º 1
0
 def add_arguments(parser: argparse.ArgumentParser):
     add_glue_args(parser)
     add_glue_inference_args(parser)
     inference_args(parser)
     create_base_args(
         parser,
         model_types=TransformerSequenceClassifier.MODEL_CLASS.keys())
Ejemplo n.º 2
0
 def add_arguments(parser: argparse.ArgumentParser):
     parser.add_argument("--data_file",
                         default=None,
                         type=str,
                         required=True,
                         help="The data file containing data for inference")
     inference_args(parser)
     create_base_args(
         parser, model_types=TransformerTokenClassifier.MODEL_CLASS.keys())