Skip to content

Commit 4792cd1

Browse files
Remove space
Co-authored-by: Carlo Lucibello <[email protected]>
1 parent 7a6c6e4 commit 4792cd1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

GNNLux/src/layers/conv.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,8 @@ y = l(g, x, w, ps, st; norm_fn = custom_norm_fn)
7474
7575
# Edge weights can also be embedded in the graph.
7676
g = GNNGraph(s, t, w)
77-
l = GCNConv(3 => 5, use_edge_weight=true)
77+
l = GCNConv(3 => 5, use_edge_weight=true)
78+
ps, st = Lux.setup(rng, l)
7879
y = l(g, x, ps, st) # same as l(g, x, w)
7980
```
8081
"""

0 commit comments

Comments
 (0)