use a single activation function by default for GNNs

BIMSBbioinfo · May 14, 2024 · a195c1f · a195c1f
1 parent f796c47
commit a195c1f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flexynesis/config.py b/flexynesis/config.py
@@ -38,6 +38,6 @@
         Real(0.0001, 0.01, prior="log-uniform", name="lr"),
         Categorical(epochs, name="epochs"),
         Integer(16, 128, name='supervisor_hidden_dim'),
-        Categorical(['relu', 'sigmoid', 'tanh', 'gelu'], name="activation")
+        Categorical(['relu'], name="activation")
     ]
 }