Skip to content

Commit 4904f91

Browse files
committed
fix output
1 parent 9b76cf5 commit 4904f91

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

GNNLux/src/layers/conv.jl

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -663,9 +663,8 @@ function (l::MegNetConv)(g, x, e, ps, st)
663663
end
664664

665665
function Base.show(io::IO, l::MegNetConv)
666-
ne = l.num_features.edge
667-
nin = l.num_features.in
668-
nout = l.num_features.out
669-
print(io, "MegNetConv(($nin, $ne) => $nout")
666+
nin = l.in_dims
667+
nout = l.out_dims
668+
print(io, "MegNetConv(", l.in_dims, " => ", l.out_dims)
670669
print(io, ")")
671670
end

0 commit comments

Comments
 (0)