From 599ba9a28c401b933fe14dc0ae149c46a66e6095 Mon Sep 17 00:00:00 2001 From: Rishabh <59335537+rbSparky@users.noreply.github.com> Date: Mon, 30 Sep 2024 16:57:11 +0530 Subject: [PATCH] Update conv_tests.jl --- GNNLux/test/layers/conv_tests.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/GNNLux/test/layers/conv_tests.jl b/GNNLux/test/layers/conv_tests.jl index 5b1cc5cc6..bf3b3b338 100644 --- a/GNNLux/test/layers/conv_tests.jl +++ b/GNNLux/test/layers/conv_tests.jl @@ -13,7 +13,7 @@ l = TransformerConv((6, ein) => 8, heads = 2, gating = true, bias_qkv = true) test_lux_layer(rng, l, g, x, outputsize = (16,), e = e, container = true) - l = TransformerConv((16, ein) => 8, heads = 2, concat = false, skip_connection = true) + l = TransformerConv((16, ein) => 16, heads = 2, concat = false, skip_connection = true) test_lux_layer(rng, l, g, x, outputsize = (16,), e = e, container = true) end