From cbe5a0db6ae5ce2dae6383d031cdbd56b97f1874 Mon Sep 17 00:00:00 2001 From: PatReis Date: Fri, 8 Sep 2023 22:01:08 +0200 Subject: [PATCH] continue keras core integration --- training_core/train_graph.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/training_core/train_graph.py b/training_core/train_graph.py index 40554335..f64f6712 100644 --- a/training_core/train_graph.py +++ b/training_core/train_graph.py @@ -21,7 +21,7 @@ parser = argparse.ArgumentParser(description='Train a GNN on a Molecule dataset.') parser.add_argument("--hyper", required=False, help="Filepath to hyperparameter config file (.py or .json).", default="hyper/hyper_esol.py") -parser.add_argument("--category", required=False, help="Graph model to train.", default="Schnet") +parser.add_argument("--category", required=False, help="Graph model to train.", default="GCN") parser.add_argument("--model", required=False, help="Graph model to train.", default=None) parser.add_argument("--dataset", required=False, help="Name of the dataset.", default=None) parser.add_argument("--make", required=False, help="Name of the class for model.", default=None)