189c2c0907cdff54f733299a93360ddfcff4ccb7,examples/pytorch/model_zoo/chem/property_prediction/regression.py,,,#,80

Before Change

                        help="number of epochs",
                        default=250,
                        type=int)
    device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
    args = parser.parse_args()
    train(args.model, args.epochs, device)

After Change

    parser = argparse.ArgumentParser(description="Molecule Regression")
    parser.add_argument("-m", "--model", type=str, choices=["MPNN", "SCHNET", "MGCN"],
                        help="Model to use")
    parser.add_argument("-d", "--dataset", type=str, choices=["Alchemy"],
                        help="Dataset to use")
    args = parser.parse_args().__dict__
    args["exp"] = "_".join([args["model"], args["dataset"]])

Instances