standard value change

Aske-Rosted · Oct 10, 2023 · 5fe093e · 5fe093e
1 parent feff375
commit 5fe093e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/graphnet/models/gnn/rnn_dynedge.py b/src/graphnet/models/gnn/rnn_dynedge.py
@@ -19,8 +19,8 @@ def __init__(
         nb_inputs: int,
         *,
         nb_neighbours: int = 8,
-        RNN_layers: int = 2,
-        RNN_hidden_size: int = 128,
+        RNN_layers: int = 3,
+        RNN_hidden_size: int = 256,
         features_subset: Optional[Union[List[int], slice]] = None,
         dynedge_layer_sizes: Optional[List[Tuple[int, ...]]] = None,
         post_processing_layer_sizes: Optional[List[int]] = None,
@@ -61,6 +61,10 @@ def __init__(
                     512,
                     1024,
                 ),
+                (
+                    1024,
+                    1024,
+                ),
                 (
                     1024,
                     512,