From 4904f911e2bf45437067aa390238062a9633d74e Mon Sep 17 00:00:00 2001 From: rbSparky Date: Sun, 4 Aug 2024 17:56:45 +0530 Subject: [PATCH] fix output --- GNNLux/src/layers/conv.jl | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl index 279d2beb3..7ea85e0a5 100644 --- a/GNNLux/src/layers/conv.jl +++ b/GNNLux/src/layers/conv.jl @@ -663,9 +663,8 @@ function (l::MegNetConv)(g, x, e, ps, st) end function Base.show(io::IO, l::MegNetConv) - ne = l.num_features.edge - nin = l.num_features.in - nout = l.num_features.out - print(io, "MegNetConv(($nin, $ne) => $nout") + nin = l.in_dims + nout = l.out_dims + print(io, "MegNetConv(", l.in_dims, " => ", l.out_dims) print(io, ")") end \ No newline at end of file