From 5c0c5a8eef09b0ce6e15ef59c8d21296738cf66e Mon Sep 17 00:00:00 2001 From: rbSparky Date: Thu, 8 Aug 2024 05:45:03 +0530 Subject: [PATCH] fix --- GNNLux/src/layers/conv.jl | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl index e66cdfcbd..d1f839c9d 100644 --- a/GNNLux/src/layers/conv.jl +++ b/GNNLux/src/layers/conv.jl @@ -637,8 +637,9 @@ end aggr::A end -# 'mean' not defined -#MEGNetConv(ϕe, ϕv; aggr = mean) = MEGNetConv(ϕe, ϕv, aggr) +function MEGNetConv(in_dims::Int, out_dims::Int, ϕe::TE, ϕv::TV; aggr::A = mean) where {TE, TV, A} + return MEGNetConv{TE, TV, A}(in_dims, out_dims, ϕe, ϕv, aggr) +end function MEGNetConv(ch::Pair{Int, Int}; aggr = mean) nin, nout = ch @@ -648,7 +649,7 @@ function MEGNetConv(ch::Pair{Int, Int}; aggr = mean) ϕv = Chain(Dense(nin + nout, nout, relu), Dense(nout, nout)) - return MEGNetConv(nin, nout, ϕe, ϕv; aggr) + return MEGNetConv(nin, nout, ϕe, ϕv, aggr=aggr) end function (l::MEGNetConv)(g, x, e, ps, st)