continue keras core integration

aimat-lab · Sep 8, 2023 · cbe5a0d · cbe5a0d
1 parent 4fb3044
commit cbe5a0d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)