From 7f1a07a5f9b7e8d8af54701e822fe53083b4e8d4 Mon Sep 17 00:00:00 2001 From: Rishabh <59335537+rbSparky@users.noreply.github.com> Date: Sun, 4 Aug 2024 17:19:44 +0530 Subject: [PATCH] Update conv.jl --- GNNLux/src/layers/conv.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/GNNLux/src/layers/conv.jl b/GNNLux/src/layers/conv.jl index 378dd61d6..cc838050e 100644 --- a/GNNLux/src/layers/conv.jl +++ b/GNNLux/src/layers/conv.jl @@ -669,7 +669,7 @@ end LuxCore.outputsize(d::NNConv) = (d.out_dims,) -function Base.show(io::IO, l::GINConv) +function Base.show(io::IO, l::NNConv) print(io, "NNConv($(l.nn)") print(io, ", $(l.ϵ)") l.σ == identity || print(io, ", ", l.σ)