fixes

JuliaGraphs · Aug 1, 2024 · 2f3f76f · 2f3f76f
1 parent f2534dd
commit 2f3f76f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/layers/conv.jl b/src/layers/conv.jl
@@ -520,7 +520,7 @@ end
 function GatedGraphConv(dims::Int, num_layers::Int;
                         aggr = +, init = glorot_uniform)
     w = init(dims, dims, num_layers)
-    gru = GRUCell(out_ch, dims)
+    gru = GRUCell(dims => dims)
     GatedGraphConv(w, gru, dims, num_layers, aggr)
 end