Skip to content

Commit b8c4db6

Browse files
authored
Update conv.jl
1 parent 01767ee commit b8c4db6

File tree

1 file changed

+1
-13
lines changed

1 file changed

+1
-13
lines changed

GNNLux/src/layers/conv.jl

Lines changed: 1 addition & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -667,20 +667,8 @@ function (l::GCNConv)(g, x, edge_weight, ps, st)
667667
return y, stnew
668668
end
669669

670-
function LuxCore.initialparameters(rng::AbstractRNG, l::NNConv)
671-
weight = l.init_weight(rng, l.out_dims, l.in_dims)
672-
if l.use_bias
673-
bias = l.init_bias(rng, l.out_dims)
674-
return (; weight, bias)
675-
else
676-
return (; weight)
677-
end
678-
end
679-
680-
LuxCore.parameterlength(l::NNConv) = l.use_bias ? l.in_dims * l.out_dims + l.out_dims : l.in_dims * l.out_dims # nn wont affect this right?
681670
LuxCore.outputsize(d::NNConv) = (d.out_dims,)
682671

683-
684672
function Base.show(io::IO, l::GINConv)
685673
print(io, "NNConv($(l.nn)")
686674
print(io, ", $(l.ϵ)")
@@ -689,4 +677,4 @@ function Base.show(io::IO, l::GINConv)
689677
l.add_self_loops || print(io, ", add_self_loops=false")
690678
!l.use_edge_weight || print(io, ", use_edge_weight=true")
691679
print(io, ")")
692-
end
680+
end

0 commit comments

Comments
 (0)