diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl index d1f839c9d..a4fe957e9 100644 --- a/GNNLux/src/layers/conv.jl +++ b/GNNLux/src/layers/conv.jl @@ -655,7 +655,7 @@ end function (l::MEGNetConv)(g, x, e, ps, st) ϕe = StatefulLuxLayer{true}(l.ϕe, ps.ϕe, _getstate(st, :ϕe)) ϕv = StatefulLuxLayer{true}(l.ϕv, ps.ϕv, _getstate(st, :ϕv)) - m = (; ϕe, ϕv, l.residual, l.num_features) + m = (; ϕe, ϕv) return GNNlib.megnet_conv(m, g, x, e), st end