2023/10/01-14:57:57 (Linux cray unknown)

pbenner · Oct 1, 2023 · 69ea2b0 · 69ea2b0
1 parent db13c10
commit 69ea2b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/coordinationnet/model_gnn.py b/coordinationnet/model_gnn.py
@@ -31,7 +31,7 @@ def __init__(self,
         # Specify model components
         model_config = DefaultGraphCoordinationNetConfig,
         # Options for dense layers
-        layers, **kwargs):
+        layers = [512, 128, 64, 1], **kwargs):
 
         super().__init__()