Fix: LayerConfig.inner_dim default handling (#8486)

#8484
pyg-team · Nov 30, 2023 · f24d3a0 · f24d3a0
1 parent 8833d0d
commit f24d3a0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_geometric/graphgym/models/layer.py b/torch_geometric/graphgym/models/layer.py
@@ -237,7 +237,7 @@ class MLP(torch.nn.Module):
     """
     def __init__(self, layer_config: LayerConfig, **kwargs):
         super().__init__()
-        if layer_config.dim_inner:
+        if layer_config.dim_inner is None:
             dim_inner = layer_config.dim_in
         else:
             dim_inner = layer_config.dim_inner