feat: more generic naming for critic, since messager will use the same

2026-04-02 18:11:49 +00:00 · 2026-04-02 18:11:49 +00:00 · 40c7e32344
commit 40c7e32344
parent 729f122be4
2 changed files with 4 additions and 3 deletions
--- a/src/MLPs/mlps.py
+++ b/src/MLPs/mlps.py
@ -21,7 +21,7 @@ class SemiGenericNetwork(nn.Module):
        return x


-class Critic(nn.Module):
+class OneDenseLayerMLP(nn.Module):
    @nn.compact
    def __call__(self, x):
        return nn.Dense(1, kernel_init=orthogonal(1), bias_init=constant(0.0))(x)