fixup: clarify GAT-specific parameters

ORNL · Mar 17, 2022 · 89fb9a0 · 89fb9a0
1 parent 388e216
commit 89fb9a0
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/hydragnn/models/create.py b/hydragnn/models/create.py
@@ -104,9 +104,12 @@ def create_model(
         )
 
     elif model_type == "GAT":
+        # FIXME: expose options to users
+        heads = 6
+        negative_slope = 0.05
         model = GATStack(
-            6,
-            0.05,
+            heads,
+            negative_slope,
             input_dim,
             hidden_dim,
             output_dim,