fix

JuliaGraphs · Aug 8, 2024 · 79f3115 · 79f3115
1 parent 5c0c5a8
commit 79f3115
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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