Skip to content

Commit 05dc33f

Browse files
fix cora example
1 parent 6239b48 commit 05dc33f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/cora.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,8 +54,8 @@ function train(; kws...)
5454
nin, nhidden, nout = size(X,1), args.nhidden, data.num_classes
5555

5656
## DEFINE MODEL
57-
model = GNNGraph(GCNConv(nin => nhidden, relu),
58-
Dropout(0.5)
57+
model = GNNChain(GCNConv(nin => nhidden, relu),
58+
Dropout(0.5),
5959
GCNConv(nhidden => nhidden, relu),
6060
Dense(nhidden, nout)) |> device
6161

0 commit comments

Comments
 (0)