Update conv_tests.jl

CarloLucibello · Sep 30, 2024 · 599ba9a · 599ba9a
1 parent a49df82
commit 599ba9a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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