95f9dc16aa2428f62b42260be603f2287332b5a8,examples/pytorch/BertNewsClassification/bert_classification.py,,,#,365

Before Change

    parser = ArgumentParser(description="Bert-News Classifier Example")

    // Add trainer specific arguments
    parser.add_argument(
        "--max-epochs", type=int, default=10, help="number of epochs to run (default: 10)"
    )
    parser.add_argument(
        "--gpus", type=int, default=0, help="Number of gpus - by default runs on CPU"
    )
    parser.add_argument(
        "--accelerator", type=str, default=None, help="Distributed Backend - (default: None)",
    )

    parser.add_argument(
        "--num-samples",
        type=int,

After Change

        help="Number of samples to be used for training and evaluation steps (default: 15000) Maximum:100000",
    )

    parser = pl.Trainer.add_argparse_args(parent_parser=parser)
    parser = BertNewsClassifier.add_model_specific_args(parent_parser=parser)
    parser = BertDataModule.add_model_specific_args(parent_parser=parser)

    mlflow.pytorch.autolog()

    args = parser.parse_args()
    dict_args = vars(args)

    if "accelerator" in dict_args:
        if dict_args["accelerator"] == "None":
            dict_args["accelerator"] = None

    dm = BertDataModule(**dict_args)
    dm.prepare_data()
    dm.setup(stage="fit")

Instances