Skip to content

Commit fb0bb1d

Browse files
authored
Update conv.jl
1 parent b8c4db6 commit fb0bb1d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

GNNLux/src/layers/conv.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -656,7 +656,7 @@ function NNConv(ch::Pair{Int, Int}, nn, σ = identity;
656656
return NNConv(nn, aggr, in_dims, out_dims, use_bias, add_self_loops, use_edge_weight, init_weight, init_bias, σ)
657657
end
658658

659-
function (l::GCNConv)(g, x, edge_weight, ps, st)
659+
function (l::NNConv)(g, x, edge_weight, ps, st)
660660
nn = StatefulLuxLayer{true}(l.nn, ps, st)
661661

662662
# what would be the order of args here?
@@ -669,7 +669,7 @@ end
669669

670670
LuxCore.outputsize(d::NNConv) = (d.out_dims,)
671671

672-
function Base.show(io::IO, l::GINConv)
672+
function Base.show(io::IO, l::NNConv)
673673
print(io, "NNConv($(l.nn)")
674674
print(io, ", $(l.ϵ)")
675675
l.σ == identity || print(io, ", ", l.σ)

0 commit comments

Comments
 (0)