wip

JuliaGraphs · Aug 7, 2024 · d5cfb7b · d5cfb7b
1 parent 4904f91
commit d5cfb7b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/GNNLux/src/GNNLux.jl b/GNNLux/src/GNNLux.jl
@@ -30,7 +30,7 @@ export AGNNConv,
        GINConv,
        # GMMConv,
        GraphConv,
-       # MEGNetConv,
+       #MEGNetConv,
        # NNConv,
        # ResGatedGraphConv,
        # SAGEConv,

diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl
@@ -650,11 +650,8 @@ function MEGNetConv(ch::Pair{Int, Int}; aggr = mean)
     return MEGNetConv(nin, nout, ϕe, ϕv; aggr)
 end
 
-
 LuxCore.outputsize(l::MegNetConv) = (l.num_features.out,)
 
-(l::MegNetConv)(g, x, ps, st) = l(g, x, nothing, ps, st) # check
-
 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))    
@@ -665,6 +662,6 @@ end
 function Base.show(io::IO, l::MegNetConv)
     nin = l.in_dims
     nout = l.out_dims
-    print(io, "MegNetConv(", l.in_dims, " => ", l.out_dims)
+    print(io, "MegNetConv(", nin, " => ", nout)
     print(io, ")")
 end