diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl index bb65fc4b0..bca5eef7d 100644 --- a/GNNLux/src/layers/conv.jl +++ b/GNNLux/src/layers/conv.jl @@ -666,20 +666,9 @@ function (l::NNConv)(g, x, edge_weight, ps, st) return y, stnew end -function LuxCore.initialparameters(rng::AbstractRNG, l::NNConv) - weight = l.init_weight(rng, l.out_dims, l.in_dims) - if l.use_bias - bias = l.init_bias(rng, l.out_dims) - return (; weight, bias) - else - return (; weight) - end -end - -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? LuxCore.outputsize(d::NNConv) = (d.out_dims,) -function Base.show(io::IO, l::GINConv) +function Base.show(io::IO, l::NNConv) print(io, "NNConv($(l.nn)") print(io, ", $(l.ϵ)") l.σ == identity || print(io, ", ", l.σ) @@ -687,4 +676,4 @@ function Base.show(io::IO, l::GINConv) l.add_self_loops || print(io, ", add_self_loops=false") !l.use_edge_weight || print(io, ", use_edge_weight=true") print(io, ")") -end \ No newline at end of file +end